hdu1597 find the nth digit（二分查找）

find the nth digit

Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 13674    Accepted Submission(s): 4194

Problem Description

假设：
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
  
  
  
  


  
  

首先可以知道一个规律，

1,2,3,4,5,6,7，。。。

第i个串一共有（1+i）*i/2

由于该序列单调递增，可以使用二分法

#include <bits/stdc++.h>
using namespace std;
typedef long long  ll;
ll f(ll x)
{
    return (1+x)*x/2;
}
int main()
{
    ll t;
    cin>>t;
    while(t--){
        ll n;
        cin>>n;
        ll l=0,r=n,mid;
        while(l<r){
            mid=l+(r-l)/2;
//            cout<<mid<<"　"<<f(mid)<<" "<<l<<endl;
            if(f(mid)>=n)
                r=mid;
            else
                l=mid+1;
        }
        l--;
//        cout<<"l:  "<<l<<"   f[l]  "<<f(l)<<endl;
//        cout<<n-f(l)<<endl;
        ll temp=n-f(l);
        int ans=temp%9;
        if(ans==0)
            ans=9;
//        cout<<"ans没除以9: "<<temp<<endl;
        cout<<ans<<endl;
    }
    return 0;
}