[프로그래머스/C++ 문제 풀이] Lv. 0 - 배열 만들기 2

문제 설명

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

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

제한 사항

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

입출력 예

l	r	result
10	20	[-1]
5	555	[5, 50, 55, 500, 505, 550, 555]

 

입출력 예 #1

• 5 이상 555 이하의 0과 5로만 이루어진 정수는 작은 수부터 5, 50, 55, 500, 505, 550, 555가 있습니다. 따라서 [5, 50, 55, 500, 505, 550, 555]를 return 합니다.

입출력 예 #2

• 10 이상 20 이하이면서 0과 5로만 이루어진 정수는 없습니다. 따라서 [-1]을 return 합니다.

---

문제 풀이

저는 이번 문제를 보며 우선 0과 5로 이루어진 수의 규칙을 찾으려고 노력했습니다.

아무리 생각해봐도 규칙을 찾을 수 없었으며, 모든 수를 하나씩 5인지 0인지 체크하는 것밖에 없다고 생각하였습니다.

그래서 매번 나눠주며 숫자로 연산하는 것보다 쉽게 숫자를 문자열로 만들어 순회하며 체크하는 방식으로 문제를 해결하였습니다.

정답 코드

풀이 시간  5m 32s

#include <string>
#include <vector>

using namespace std;

// 문자열이 전부 0 또는 5로만 이루어졌는지 확인하기 위한 함수
bool CheckNum(string str)
{
    bool isResult = true;

    for (char num : str)
    {
        if (num != '0' && num != '5')
        {
            isResult = false;
            break;
        }
    }

    return isResult;
}

vector<int> solution(int l, int r) {
    vector<int> answer;

    for (int i = l; i <= r; i++)
    {
        // 숫자를 문자열로 변환
        string numStr = to_string(i);

        // 해당 문자열이 전부 0 또는 5로 이루어진 문자열인지 체크
        if (CheckNum(numStr))
        {
            answer.push_back(i);
        }
    }

    // 비어있는 경우 -1 넣기
    if (answer.empty())
        answer.push_back(-1);

    return answer;
}