Kyselyt

input/code.cpp: In function 'int num(long int)':
input/code.cpp:35:1: warning: control reaches end of non-void function [-Wreturn-type]
 }
 ^

#include <iostream>
#include <math.h>

using namespace std;


int num(long q)
{
    if( q <= 9)
    {
        return q;
    }

    long sum = 9;
    long n_sum;
    for(int i=2; i < 18; i++)
    {
        n_sum = i * 9 * pow(10, i-1) + sum;
        if(q <= n_sum)
        {
            if(q%i == (sum + 1) % i)
            {
                return (( (q-sum) % (i * (long)(pow(10, i))) ) / (pow(10, i-1)*i)) + 1;
            }
            for(int j = 1; j < i; j++)
            {
                if(q%i == (sum + j + 1) % i)
                {
                    return ( (q-sum-j) % ((long)(pow(10, i-j)*i)) ) / (pow(10, i-j-1)*i);
                }
            }
        }
        sum = n_sum;
    }
}

int main()
{
    long q;
    int n;
    cin >> n;
    for (int i = 0; i < n; i++)
    {
        cin >> q;
        cout << num(q) << "\n";
    }
}

Test 1

Group: 1

Verdict: ACCEPTED

Test 2

Group: 2

Verdict: ACCEPTED

Test 3

Group: 3

Verdict: WRONG ANSWER