hdu 1597 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): 8356    Accepted Submission(s): 2383

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
  
  

 

AC代码：

#include<stdio.h>
#include<iostream>
using namespace std;
int main ()
{
	int T,n;
	int i,j;
	for(i=1;i<n;i++)		
             n-=i;
	if(n%9)
		cout<<n%9<<endl;
	else
		cout<<9<<endl;
			
	}
   return 0;
}

自己写的代码，只能处理小范围的数据

#include<stdio.h>
#include<iostream>
using namespace std;
int main ()
{
	int T,n;
	int i,j,k;
	__int64 a[10000],sum[10000];
	memset(sum,0,sizeof(sum));
	
	for(i=1;i<=10000;i++)
		sum[i]=(1+i)*i/2;		
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		for(i=1;i<=10000;i++)
		{
			if(sum[i]<n && n<sum[i+1])
			{
				k=(n-sum[i])%9;
				break;
			}
			if(sum[i]==n)
			{
				k=i%9;
				break;
			}
		}
		if(k==0)
			cout<<9<<endl;
		else
			cout<<k<<endl;
		
	}
	return 0;
}