【题解】TES-Intelligence Test

【题解】\(TES-Intelligence\) \(Test\)

逼自己每天一道模拟题

传送：\(TES-Intelligence\) \(Test\) \([POI2010]\) \([P3500]\)

【题目描述】

\(Byteotian\) 智力测试机构的一项工作是按照一定的规则删除一个序列的数字，得到一个确定的数列。\(Byteasar\) 很渴望成为 \(Byteotian\) 智力测试机构的主管，但是他在这个工作上做的并不好，但毕竟熟能生巧，他打算做大量的练习，所以他希望你写一个程序来快速判断是否正确。

【输入】

第一行一个整数 \(n\)，第二行 \(n\) 个整数\(a_i\)，表示最初的序列，第三行一个整数 \(T\)，表示 \(T\) 个 \(Byteasar\) 经过一系列删除得到的序列，每个序列两行，第一行给出长度 \(m\)，然后下一行为 \(m\) 个整数 \(b_i\)。

【输出】

共 \(m\) 行，如果 \(Byteasar\) 的序列是由最初的序列删除一些数后得到的，那么输出 \(TAK\)，否则输出 \(NIE\)。

【样例】

样例输入：
7
1 5 4 5 7 8 6
4
5
1 5 5 8 6
3
2 2 2
3
5 7 8
4
1 5 7 4

样例输出：
TAK
NIE
TAK
NIE

【数据范围】

\(100\%\) \(1 \leqslant n,T \leqslant 1e6\) \(,\) \(1 \leqslant a_i,b_i​ \leqslant 1e6\) \(,\) \(1 \leqslant m_i \leqslant n\)

---

【分析】

首先用一个 \(vector\) 将原数列中出现了的各数值所在的位置记录下来，然后每读一个序列进来，就扫一遍，对于每一个扫到的数，都在 \(vector\) 中进行一次二分，找到最靠前的那一个和它相同的数，然后将这个位置作为下一次二分查找的起点。

【Code】

#include<vector>
#include<cstdio>
#define Re register int
#define F(a,b) for(Re i=a;i<=b;++i)
const int N=1e6+5;
std::vector<int>a[N];
int n,m,T,x,st,b[N];
inline void in(Re &x){
    int fu=0;x=0;char c=getchar();
    while(c<'0'||c>'9')fu|=c=='-',c=getchar();
    while(c>='0'&&c<='9')x=(x<<1)+(x<<3)+(c^48),c=getchar();
    x=fu?-x:x;
}
inline void sakura(Re &flag,Re now,Re l,Re r){//二分查找
    while(l<r){
        Re mid=l+r>>1;
        if(a[now][mid]>st)r=mid;
        else l=mid+1;
    }
    flag|=a[now][l]>st,st=a[now][l];//最终判断。更新st。
}
inline int judge(){//判断这个序列是否合格
    F(1,m){
        Re flag=0,now=b[i],l=0,r=a[now].size();
        if(!r)return 0;//原序列中没有这个数
        --r;
        sakura(flag,now,l,r);//二分
        if(!flag)return 0;//二分的结果，看st后面是否存在当前扫到的这个数
    }
    return 1;
}
int main(){
    in(n);F(1,n)in(x),a[x].push_back(i);
    in(T);
    while(T--){
        in(m);st=0;
        F(1,m)in(b[i]);
        puts(judge()?"TAK":"NIE");
    }
}