Description
霸中智力测试机构的一项工作就是按照一定的规则删除一个序列的数字,得到一个确定的数列。Lyx很渴望成为霸中智力测试机构的主管,但是他在这个工作上做的并不好,俗话说熟能生巧,他打算做很多练习,所以他希望你写一个程序来快速判断他的答案是否正确。
Input
第一行为一个整数m(1<=m<=1000000)第二行包括m个用空格分开的整数ai(1<=ai<=1000000),组成了最初的序列,第三行为一个整数n(1<=n<=1000000),表示n个Lyx经过一系列删除得到的序列,每个序列两行,第一行给出长度L(1<=L<=m),然后下一行为L个由空格分开的整数bi(1<=bi<=1000000)。
Output
共n行,如果Lyx的序列确实是由最初的序列删除一些数得到,就输出TAK,否则输出NIE。
Solution
无脑题。
下文s为查询序列。
明显可知在原序列中贪心取s每个元素,越靠前越好。
显而易见选了前面的能够选择后继的区间比选了后面的更广。
直接二分最前面且大于las位置的即可。
Code
#include<bits/stdc++.h>
using namespace std;
int a[1000010],b[1000010],n,q,l;
vector<int>v[1000010];
int main(){
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d",&a[i]);
v[a[i]].push_back(i);
}
scanf("%d",&q);
for(int i=1;i<=q;i++){
int las=0;
bool flag=1;
scanf("%d",&l);
for(int j=1;j<=l;j++){
scanf("%d",&b[j]);
if(v[b[j]].size()==0){
for(int k=j+1;k<=l;k++)
scanf("%d",&b[j]);
flag=false;
break;
}
int p=upper_bound(v[b[j]].begin(),v[b[j]].end(),las)-v[b[j]].begin();
if(p==v[b[j]].size()){
for(int k=j+1;k<=l;k++)
scanf("%d",&b[j]);
flag=false;
break;
}
las=v[b[j]][p];
}
if(flag) printf("TAK\n");
else printf("NIE\n");
}
}