第三次模拟测试-C

题目描述：
咕咕东 正在上可怕的复变函数，但对于稳拿A Plus的 咕咕东 来说，她早已不再听课。
此时她在睡梦中突然想到了一个奇怪的无限序列：112123123412345…
这个序列由连续正整数组成的若干部分构成，其中第一部分包含1至1之间的所有数字，第二部分包含1至2之间的所有数字，第三部分包含1至3之间的所有数字，第i部分总是包含1至i之间的所有数字。
所以，这个序列的前56项会是11212312341234512345612345671234567812345678912345678910，其中第1项是1，第3项是2，第20项是5，第38项是2，第56项是0。
咕咕东 现在想知道第 k 项数字是多少！但是她睡醒之后发现老师讲的东西已经听不懂了，因此她把这个任务交给了你。
输入：
输入由多行组成。
第一行一个整数q表示有q组询问(1<=q<=500)
接下来第i+1行表示第i个输入ki，表示询问第ki项数字。(1<=k_i<=10^{18})
输出：
输出包含q行
第i行输出对询问ki的输出结果。
sample：
input：
5
1
3
20
38
56
output：
1
2
5
2
0
题目分析：
考场上写了半天忽然发现它对应输出的是一个一个的数字，也就是10表示两个数1和0，且1和0各占一个位置，那么到了第56个数字就是0而不是10，所以我一开始只得了30分（就是56以内的）
那么考虑到题目给的范围非常大，那么我们就得用数学规律来解决了。首先既然他考虑的是每一个数每一位上的数，那么我们就按照数的位数来分类。这样计算一位数有多少个数字、两位数有多少个数字会比较容易。
把它看成一个梯形、

那么第i个区域内，每一层的元素个数，当前块的上底下底和总共的字符串数目就能求出来了

    for(int i=1; ;i++)
	{
        len=i; 
		num[i]=i*height+num[i-1];//上一层+位数*当前位数的数有多少
        up[i]=down[i-1]+i;//上底就是下底+位数
        down[i]=up[i]+i*(height-1);//下底就是上底+位数*当前位数的数有多少（要减一）
        sum[i]=sum[i-1]+((up[i]+down[i])*height)/2;//看成题型求和
        if(sum[i]>=1e18)//k的取值就是1e18 
		    break;
        height=height*10;//下一层继续
    }

那这样对于每一个输入的k来说，我们就能逐步缩小范围进行查找了，首先确定他在哪个梯形之中，然后确定他在哪一行，之后就能求出他是那个数字了。
首先确定梯形

        for (int i=1; i<=len; i++)
		{
            if(k<=sum[i])
			{

然后要确定他在梯形的哪一行之中，如果直接搜索会TLE，所以我们采用二分搜索的方法，求出二分的时候所占有的数然后与k比较，如果大了就往前小了就往后

                k=k-sum[i-1];
                height=((down[i]-up[i])/i)+1;
                long long left=1,right=height,level=0;
                while(left+1<right)
				{
                    long long mid=(left+right)/2;
                    long long temp=((up[i]+(up[i]+i*(mid-1)))*mid)/2;
                    if(temp>k) 
					    right=mid;
                    else 
					    left=mid;
                }

当然由于层数是离散的，所以我们找到了之后要确定k所在的层数是第几层，那就找附近的两个比较，看谁小

                if(((up[i]+(up[i]+i*(left-1)))*left)/2<k) 
				    level=right;
                else 
				    level=left;

确定了层之后就可以找在哪个地方了，注意不要直接从前往后搜索，否则TLE。这样和上面一个步骤，看看首先他在几位数里面，这个和上面是一样的。

                k=k-((up[i]+(up[i]+i*(level-2)))*(level-1))/2;
                for (int j=1; j<=i; j++)
				{
                    if(k<=num[j])

然后求出这几位数的第几位数，之后就可以求出要的数字了

                        k=k-num[j-1];
                        int remainder;
                        if(k%j==0)
                            remainder=j;
                        else
                            remainder=k%j;
                        k=k/j;
                        int theNumber=k+pow(10,j-1)-1;
                        int temp=(int)log10(theNumber)+1-remainder;

找到了就求它的个位数就可以了，个位数就是我们所要的答案

                        while(temp--) 
						{
							theNumber=theNumber/10;
						}
                        cout<<theNumber%10<<endl;

代码如下：

#include <iostream>
#include <cmath>
#include <cstring>
using namespace std;
long long sum[20],up[20],down[20],num[20],height=9;
int main()
{
	memset(sum,0,sizeof(sum));
	memset(up,0,sizeof(up));
	memset(down,0,sizeof(down));
	memset(num,0,sizeof(num));
	int len;
    for(int i=1; ;i++)
	{
        len=i; 
		num[i]=i*height+num[i-1];
        up[i]=down[i-1]+i;
        down[i]=up[i]+i*(height-1);
        sum[i]=sum[i-1]+((up[i]+down[i])*height)/2;
        if(sum[i]>=1e18) 
		    break;
        height=height*10;
    }
    int q;
    cin>>q;;
    for(int ii=1;ii<=q;ii++)
	{
		long long k;
        cin>>k;
        for (int i=1; i<=len; i++)
		{
            if(k<=sum[i])
			{
                k=k-sum[i-1];
                height=((down[i]-up[i])/i)+1;
                long long left=1,right=height,level=0;
                while(left+1<right)
				{
                    long long mid=(left+right)/2;
                    long long temp=((up[i]+(up[i]+i*(mid-1)))*mid)/2;
                    if(temp>k) 
					    right=mid;
                    else 
					    left=mid;
                }
                if(((up[i]+(up[i]+i*(left-1)))*left)/2<k) 
				    level=right;
                else 
				    level=left;
                k=k-((up[i]+(up[i]+i*(level-2)))*(level-1))/2;
                for (int j=1; j<=i; j++)
				{
                    if(k<=num[j])
					{
                        k=k-num[j-1];
                        int remainder;
                        if(k%j==0)
                            remainder=j;
                        else
                            remainder=k%j;
                        k=k/j;
                        int theNumber=k+pow(10,j-1)-1;
                        int temp=(int)log10(theNumber)+1-remainder;
                        while(temp--) 
						{
							theNumber=theNumber/10;
						}
                        cout<<theNumber%10<<endl;
                        break;
                    }
                }
                break;
            }
        }
    }
    return 0;
}