Step5. 专题 1.ACM模拟 HDU1597 find the nth digit【模拟】

http://acm.hdu.edu.cn/showproblem.php?pid=1597

先查找是第几行，然后对9取余即可。

注意，i从1开始寻找时，代码虽然简介，但是不好，有风险。查找的过程耗时太大。

代码1（模拟的方式，不推荐）

这两次提交的是代码1，第二次提交还加了cin 优化，结果时间却更长了。

#include<iostream>
#include<queue>
#include<cstring>
#define ll long long unsigned 
using namespace std;
int main(){	
	int T;cin>>T;
	while(T--){
		ll n;cin>>n;
		ll i=1;//代表第几行 
		//n^(1/2)查找方式 
		while(i*(i+1)<2*n)i++;
		int x=(2*n+i-i*i)/2;
		x%=9;if(x==0)x=9;
		cout<<x<<endl;
	} 
	return 0;
} 

代码2（使用公式，推荐）

对于第i（i从1开始）行，最后一个元素索引为i（i+1） 且满足i（i+1）/2  >=  n

所以i至少大于等于（2n）^(0.5)。从这个点开始寻找即可。时间上缩短了一个数量级。

#include<iostream>
#include<queue>
#include<cstring>
#include<cmath>
#define ll long long unsigned 
using namespace std;
int main(){	

	int T;cin>>T;
	while(T--){
		ll n;cin>>n;
		ll i=1;//代表第几行 
		//查找方式，遍历次数在10以内
		for(i=pow(2*n,0.5);i*(i+1)<2*n;i++){}
		int x=(2*n+i-i*i)/2;
		x%=9;if(x==0)x=9;
		cout<<x<<endl;
	} 
	return 0;
}