[백준 1790번] 수 이어 쓰기 2 (C++)

문제링크 : https://www.acmicpc.net/problem/1188

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int main() 
{
    ios::sync_with_stdio(false);
    cin.tie(NULL);

    int N, K;
    cin >> N >> K;

    int tmp = 0;
    int result = -1;
    for(int i=1; i<=N; i++)
    {
        string stri = to_string(i);
        tmp += stri.size();
        if(tmp >= K)
        {
            int gap = tmp - stri.size();
            result = stri[K-gap-1]-'0';
            break;
        }
    }

    cout << result;

    return 0;
}

 

1~N까지의 값을 모두 문자열로 변환하여 길이를 체크했다.

해당 길이 값을 누적하여 더하며 K값보다 크거나 같은 순간을 체크한다.

이후 누적하여 더해진 값과 현재 문자열의 길이 값을 통해 차이를 구해주고, 해당 차이를 참고하여 현재 문자열에 맞는 문자를 출력해준다.

주의할 점은 불가능한 경우 -1을 출력해야 한다는 것이다.