[BZOJ 2083] Poi 2010 Intelligence test · 二分

时间:2021-05-24 20:01:20

一开始以为是个神题后来发现是个大水题。。。

next[i][j]表示数字i在原序列中出现第j次的位置,now用来记录对于当前的b[i],最小的合法位置是多少。每次二分更新now,如果now>n说明在原序列中没有合法的b[i]存在,直接退出。

二分必须要手写,因为vector里用lower_bound返回的并不是下标值而是迭代器(具体是什么我也不清楚)。

PS:机房小伙伴用倍增的方法写但是超时了,并不知道为什么。二分大法好!    

#include <stdio.h>
#include <string.h>
#include <algorithm>
#include <vector>
using namespace std;
#define pb push_back

const int N=1000005;
vector<int> next[N];
int n,m,l,a[N],b[N],now;
bool can;

inline int get(){
	int p=0;char x=getchar();
	while (x<'0' || x>'9') x=getchar();
	while (x>='0' && x<='9') p=p*10+x-'0',x=getchar();
	return p;
}

inline int BS(int t,int k){
	int l=0,r=next[t].size()-1,mid,ret=n+1;
	while (l<=r) {
		mid=(l+r)>>1;
		if (next[t][mid]>k) ret=mid,r=mid-1;
			else l=mid+1; 
	}
	if (ret==n+1) return ret;
		else return next[t][ret];
}

int main(){
	n=get();
	for (int i=1;i<=n;i++){
		a[i]=get();
		next[a[i]].pb(i);
	}
	m=get();
	while (m--){
		l=get();now=0;can=1;
		for (int i=1;i<=l;i++) b[i]=get();
		for (int i=1;i<=l;i++){
			now=BS(b[i],now);//更新now 
			if (now>n) {
				can=0;
				break;
			}
		}
		if (can) printf("TAK\n");
			else printf("NIE\n");
	}
	return 0;
}