CSP-M2 C - 咕咕东的奇妙序列 T4 1216E2 二分 Gym - 270737E CF 1216E2

题目描述

时间限制 1s
空间限制 64MB
题目描述
咕咕东 正在上可怕的复变函数，但对于稳拿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个输入k_i
，表示询问第k_i项数字。(1<=k_i<=10^{18})

输出格式
输出包含q行
第i行输出对询问k_i的输出结果。

样例输入
5
1
3
20
38
56

样例输出
1
2
5
2
0

数据点 q（上限） k(上限)
1,2,3 500 55
4,5,6 100 10^6
7,8,9,10 500 10^{18}

解题思路

观察数字的规律，可以发现第 1-9 项之间，第 11-99 项 … 之中相邻项数字的数目构成等差序列，公差为序号的位数。因而根据这一特性，当题目给出要计算第 i 项数字的值时，我们可以根据等差数列的性质定位到这个数字在第几项中，然后再从这一项中运用等差数列的性质计算出要求的第 i 个数字具体的值。

因而在程序接受输入前，计算每组等差数量的前缀和，然后根据输入 i 的值判断目标序号的位数，然后从 10^位数 到 10^位数-1 中用二分法找到要求的数在哪个对应的项中，最后再找出这个数具体的值。

程序源码

#include<iostream>
#include<string>
#include<list>
#include<algorithm>
#include<string.h>
#include<cmath>
using namespace std;
//482720320
long long qianzhuihe[20] = { 0 }; //求取前缀和
long long jishuhe[11] = { 0 }; //某一级数的总和

long long qiuweishu(long long n) { //求某个数数位大小
	long long rs = 0;
	while (1) {
		n /= 10;
		rs++;
		if (n == 0) {
			break;
		}
	}
	return rs;
}

void calcSum() { //初始化计算总和
	long long a1 = 1;
	long long a2 = a1 * 10;
	long long weishu;
	for (int t1 = 1; t1 <= 10; t1++) {
		weishu = qiuweishu(a1);
		qianzhuihe[t1] = qianzhuihe[t1 - 1] + weishu * (a2 - a1);
		jishuhe[t1] = qianzhuihe[t1 - 1] * (a2 - a1) + ((a2 - a1 + 1) * (a2 - a1) / 2) * weishu;
		a1 *= 10;
		a2 = a1 * 10;
	}
}

long long getmin(long long weishu) { //返回某一位数的最小值
	long long rs = 1;
	for (long long i = 1; i < weishu; i++) rs *= 10;
	return rs;
}

long long getrs(long long n) { //求出序列中第n位的字符
	long long level = 0;
	while (n > qianzhuihe[level]) { //判断属于哪一级
		level++;
	}
	n -= qianzhuihe[level - 1]; //减去对应的前缀数量
	long long min = getmin(level);
	long long pianyi = (n - 1) / level; //定位到字符
	long long mubiao = min + pianyi;
	long long weishu = (n - 1) % level;
	string s(to_string(mubiao));
	return s[weishu]; //返回结果
}

long long erfenzhi(long long& mid, long long& level, long long& min) { //返回第 min 和 mid 之间项数总和
	return qianzhuihe[level - 1] * (mid - min + 1) + ((1 + mid - min + 1) * (mid - min + 1) / 2) * level;
}
long long check(long long n) { //第一步处理，定位 n 属于哪一行
	long long level = 0;
	while (n > jishuhe[level]) { //判断 n 的位数
		n -= jishuhe[level];
		level++;
	}

	//二分查找，找到大于等于 n 的 mid 值
	long long min = getmin(level);
	long long max = min * 10 - 1;

	long long min2 = min, max2 = max;
	long long mid = 0;
	while (min2 != max2) {
		long long mid = (min2 + max2) / 2;
		if (n <= erfenzhi(mid, level, min)) {
			max2 = mid;
		}
		else {
			min2 = mid + 1;
		}
	}
	mid = min2;
	if (mid == min) {
		return getrs(n); //第二部处理
	}
	else { //还要减去前面项数的和，才能进行第二部处理
		long long no = qianzhuihe[level - 1] * (mid - min) + ((1 + mid - min) * (mid - min) / 2 * level);
		return getrs(n - no);
	}

}
int main()
{
	calcSum();
	long long n, a, b;

	cin >> n;

	for (long long ni = 0; ni < n; ni++) { //总共 n 次查询
		cin >> a;
		cout << check(a) - '0' << endl;
	}

	return 0;
}