51nod3593 小明的数字表

3593 小明的数字表

小明现在爱上了查询，他遇到了这样一个查询问题，给出 n 个数 (n<=100000，
有 q 个查询 (q<=100000)，每组查询 3 个数：u v w，表示查找第 u 位 (个位为第1位，十位为第2位，依此类推)为数字 v 的所有数字中，第 w 小的数，如果这个数字不存在，输出“-1”。小明不会做这个题，聪明的你可以帮助小明解决这个问题吗？

注：不考虑所有数字中的前导 0。

输入

第一行输入一个n(n<=100000)，表示数列的长度；
第二行n个正整数ai(0<=ai<=100000000)；
第三行一个数字q(q<=100000)表示询问的次数；
下面q行，每行给出三个数字u,v,w,表示查找第u位为数字v的所有数字中，第w小的数。

输出

每个询问输出一个数字，查询不到输出“-1”。

数据范围

n<=100000
0<=ai<=100000000
q<=100000

输入样例

4
1 21 22 3
3
2 2 2
1 1 2
2 2 3

输出样例

22
21
-1

样例解释

第2位数字为2的数字包括 (21,22) 第 2 小的数字为22；

第1位数字为1的数字包括 (1,21) 第2小的数字为21；

第2位数字为2的数字包括 (21,22) 第3小的数字不存在，输出“-1”。

解析：

如果单纯的暴力做的话，对于每个询问逐个枚举这 n 个数字，去找到对应的答案，这个做法会超时。

利用空间换时间的思想，根据题目数据范围，按照每一位的情况来记录数字，我们用一个二维的 vector<int> 数组 vects 来保存预处理的数据，vects[3][5] 里面保存了所有第 3 位为 5 的数字，这样原来的一个数，可能会被保存 10 份。

我们预先对 vects中所有的元素进行排序，在询问 u,v,w 的时候，我们可以直接返回 vects[u][v] 这个 vector 的第 w 个元素即可。

放代码：

#include<bits/stdc++.h>
using namespace std;
vector<int> a[11][11];
int main(){
	int n,x;
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%d",&x);
		if(x==0){
			a[1][0].push_back(0);
		}
		int t=x;
		for(int u=1;u<=9;u++){
			while(x){
				a[u][x%10].push_back(t);
				x/=10;
				break;
			}
		}
	}
	for(int i=1;i<=9;i++){
		for(int j=0;j<=9;j++){
			sort(a[i][j].begin(),a[i][j].end());
		}

	}
	int q;
	scanf("%d",&q);
	while(q--){
		int u,v,w;
		scanf("%d%d%d",&u,&v,&w);
		if(w>a[u][v].size())    puts("-1");
		else                    printf("%d\n",a[u][v][w-1]);
	}
	return 0;
}