P3955 [NOIP2017 普及组] 图书管理员

题目描述

        图书馆中每本书都有一个图书编码，可以用于快速检索图书，这个图书编码是一个正整数。 每位借书的读者手中有一个需求码，这个需求码也是一个正整数。如果一本书的图书编码恰好以读者的需求码结尾，那么这本书就是这位读者所需要的。 小D刚刚当上图书馆的管理员，她知道图书馆里所有书的图书编码，她请你帮她写一个程序，对于每一位读者，求出他所需要的书中图书编码最小的那本书，如果没有他需要的书，请输出lns="http://www.w3.org/1998/Math/MathML">
−1。

输入

第一行，包含两个正整数
n,q，以一个空格分开，分别代表图书馆里 书的数量和读者的数量。

接下来的n行，每行包含一个正整数，代表图书馆里某本书的图书编码。

接下来的q行，每行包含两个正整数，以一个空格分开，第一个正整数代表图书馆 里读者的需求码的长度，第二个正整数代表读者的需求码。

输出

q行，每行包含一个整数，如果存在第i个读者所需要的书，则在第i行输出第i个读者所需要的书中图书编码最小的那本书的图书编码，否则输出-1

样例输入 

5 5 
2123 
1123 
23 
24 
24 
2 23 
3 123 
3 124 
2 12 
2 12

样例输出 

23 
1123 
-1 
-1 
-1 

提示

【数据规模与约定】

对于 
20%的数据，
1≤n≤2。

另有 
20%的数据，
q=1。

另有 
20%的数据，所有读者的需求码的长度均为 
1。

另有 
20%的数据，所有的图书编码按从小到大的顺序给出。

对于 
100%的数据，
1≤n≤1,000,1≤q≤1,000，所有的图书编码和需求码均 不超过 
10,000,000。

思路

先求图书编码的长度，在sort，最后再比较(用图书编码mod10的需求码次方取最后几位)，同时用数组打标记，详见代码

代码

#include <bits/stdc++.h>
using namespace std;
int n,q;
int s[1010],s2[1010];
bool b2[1010];
int len[1010],len2[1010];
bool flag;
int ws(int x)
{
	int wsu=0;
	while(x!=0)
	{
		wsu++;
		x/=10;
	}
	return wsu;
}
int main()
{
	cin>>n>>q;
	for(int i=1;i<=n;i++)
	{
		cin>>s[i];
		int l=ws(s[i]);
		len[i]=l;
	}
	for(int i=1;i<=q;i++)
	{
		cin>>len2[i]>>s2[i];
	}
	sort(s+1,s+1+n);
	sort(len+1,len+1+n);
	for(int i=1;i<=q;i++)
	{
		for(int j=1;j<=n;j++)
		{
			if(len[j]>=len2[i])
			{
				int num=pow(10,len2[i]);
				int a=s[j]%num;
				if(a==s2[i]&&b2[i]==0)
				{
					cout<<s[j]<<endl;
					b2[i]=1;
				}
			}	
		}
		if(b2[i]==0)
		{
			cout<<"-1"<<endl;
		}
	}
	return 0;
}