【bzoj2083】[Poi2010]Intelligence test

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。

Sample Input

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
Sample Output

TAK

NIE

TAK

NIE

题解
涨知识
对于每个数开vector记录位置
每次在vector中二分合法的最小位置
如果upper_bound找不到合法值则返回size()

代码

#include<bits/stdc++.h>
typedef long long ll;
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int m,n;
vector<int>a[1000005];
int b[1000005],p[1000005];
bool Judge()
{
    int n=read(),now=0;
    for (int i=1;i<=n;i++)
        b[i]=read(),p[b[i]]=0;
    for (int i=1;i<=n;i++)
    {
        p[b[i]]=upper_bound(a[b[i]].begin()+p[b[i]],a[b[i]].end(),now)-a[b[i]].begin();
        if (p[b[i]]==a[b[i]].size()) return 0;
        now=a[b[i]][p[b[i]]];  
    }
    return 1;
}
int main()
{
    m=read();
    for (int i=1;i<=m;i++)
    {
        int x=read();
        a[x].push_back(i);
    }
    n=read();
    while (n--)
    {
        if (Judge()) puts("TAK");
        else puts("NIE");
    }
    return 0;
}