CSES - Datatähti 2018 alku - Results
Submission details
Task:Kyselyt
Sender:Leiska
Submission time:2017-10-05 16:58:50 +0300
Language:C++
Status:COMPILE ERROR

Compiler report

input/code.cpp: In function 'int nthdigit(longl)':
input/code.cpp:23:35: error: expected ')' before ';' token
         x = (power(10,m-1) + (n/m);
                                   ^
input/code.cpp:25:46: error: expected ')' before ';' token
     } else x = (power(10,m-1) + (1+((n-1)/m));
                                              ^

Code

#include <iostream>

using namespace std;
typedef long long longl;

longl power(longl x, longl e) {
    if (e == 0) return 1;
    if (e == 1) return x;
    return x * power(x,e-1);
}

int nthdigit(longl k) {
    longl n, x, p, m = 1;
    
    for(;;m++) {      
        if (k-(power(10,m))-1 < 0) break;
    }
    
    n = k - (power(10,m-1)-1);  
    p = n % m;
    
    if( p==0 ) {
        x = (power(10,m-1) + (n/m);
        p = m;
    } else x = (power(10,m-1) + (1+((n-1)/m));
    
    p = -(p-(m+1));
    return ((x % power(10,p))/(power(10,p-1)));
}

int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(NULL);
    
    int q; longl k; cin >> q;
    for(;q>0; q--) {
        cin >> k;
        cout << nthdigit(k) << "\n";
    }
}