[C++] 프로그래머스 배열 만들기 2

문제설명

정수 l과 r이 주어졌을 때, l 이상 r이하의 정수 중에서
숫자 "0"과 "5"로만 이루어진 모든 정수를 오름차순으로
저장한 배열을 return 하는 solution 함수를 완성해 주세요.

만약 그러한 정수가 없다면, -1이 담긴 배열을 return 합니다.

제한사항

• 1 ≤ l ≤ r ≤ 1,000,000

입출력 예

l	r	result
5	555	[5, 50, 55, 500, 505, 550, 555]

풀이

‘0’과 ‘5’로만 이루어진 정수만 찾기 위해 이중 for문을 작성하여 문제를 풀 수 있었습니다.

#include <string>
#include <vector>

using namespace std;

vector<int> solution(int l, int r) {
    vector<int> answer;
    
    for(int i = l; i <= r; i++)
    {
        string tmp = to_string(i);
        bool chk = true;
        for(int j = 0; j < tmp.length(); j++)
        {
            if(!(tmp[j] == '0' || tmp[j] == '5'))
            {
                chk = false;
            }
        }
        if(chk)
        {
            answer.push_back(i);
        }
    }
    
    if(answer.size() == 0)
    {
        answer.push_back(-1);
    }
    
    return answer;
}

결과