HDU - 1597 find the nth digit

假设：
S1 = 1
S2 = 12
S3 = 123
S4 = 1234
.........
S9 = 123456789
S10 = 1234567891
S11 = 12345678912
............
S18 = 123456789123456789
..................
现在我们把所有的串连接起来
S = 1121231234.......123456789123456789112345678912.........
那么你能告诉我在S串中的第N个数字是多少吗？
Input输入首先是一个数字K，代表有K次询问。
接下来的K行每行有一个整数N(1 <= N < 2^31)。Output对于每个N，输出S中第N个对应的数字.
Sample Input

6
1
2
3
4
5
10

Sample Output

1
1
2
1
2
4
暴力模拟

#include<iostream>
#include<cstdio>
#include<cmath>

using namespace std;

int main()
{
    long long k,n,o;
    scanf("%lld",&k);
    while(k--)
    {
        scanf("%lld",&n);
        long long l=1,r=1LL<<31;
        long long ans=0,mid;
        while(l<=r)
        {
            mid=(l+r)/2LL;
            if((mid+1)*mid/2<n)
            {
                l=mid+1;
                ans=mid;
            }
            else
                r=mid-1;
        }
        n=n-(ans+1)*ans/2;
        o=(n-1)%9+1;
        printf("%lld\n",o);
    }

    return 0;    
}