[프로그래머스 1] 문자열 다루기 기본 (C/C++)

#include <string>
#include <vector>
using namespace std;
 
bool solution(string s) 
{
    int flag = 0;
    
    if(s.size() == 4 || s.size() == 6)
    {
        for(int i = 0; i < s.size(); i++)
        {
            if('0' <= s[i] && s[i] <= '9')
            {
                continue;
            }
            else
            {
                flag = 1;
                break;
            }
        }
 
        if(flag == 1)
        {
            return false;
        }
        else
        {
            return true;
        }   
    }
    else
    {
        return false;
    }
}