백준 1436번 영화감독 슘 [C++]

https://www.acmicpc.net/problem/1436

1436번: 영화감독 숌

 

 

666이 포함된 숫자들 중 입력으로 주어진 숫자 번 째로 작은 숫자를 찾으면 된다.

 

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>

using namespace std;

int main() {
    int n;
    cin >> n;

    vector<int> arr;
    int start = 666;
    while (arr.size() != n){
        int count = 0;
        int temp = start;
        
        while (temp > 0){
            if(temp % 10 == 6) {
                count++;
            } else {
                count = 0;
            }

            if(count == 3){
                arr.push_back(start);
            }

            temp /= 10;
        }
        start++;
        
    }
    sort(arr.begin(), arr.end());
    cout << arr[arr.size()-1];

}

666 부터 반복을 시작하여, 1씩 증가하며 차례로 666을 포함하는 숫자인지를 확인한다. 666을 포함하는 숫자라면 arr 배열에 추가하고, 주어진 입력 n과 arr의 사이즈가 같아진다면, 반복문을 종료하고 arr을 정렬한다면, arr 배열의 마지막 원소가 곧 문제의 답이 된다.