POI2010 Intelligence Test

Intelligence Test

霸中智力测试机构的一项工作就是按照一定的规则删除一个序列的数字，得到一个确定的数列。Lyx很渴望成为霸中智力测试机构的主管，但是他在这个工作上做的并不好，俗话说熟能生巧，他打算做很多练习，所以他希望你写一个程序来快速判断他的答案是否正确。

输入

第一行为一个整数m（1<=m<=1000000）第二行包括m个用空格分开的整数ai（1<=ai<=1000000），组成了最初的序列，第三行为一个整数n（1<=n<=1000000），表示n个Lyx经过一系列删除得到的序列，每个序列两行，第一行给出长度L（1<=L<=m），然后下一行为L个由空格分开的整数bi（1<=bi<=1000000）。

输出

共n行，如果Lyx的序列确实是由最初的序列删除一些数得到，就输出TAK，否则输出NIE。

比较水的一题
设后来的序列为B
为了让B尽可能是由A删除得到的
所以B中每个数在A中的位置都要尽可能靠前

为了实现它可以把A中每个数字出现的位置都用vector存一下,
所以如果要知道B中的一个数可以出现的最前位置只需要在这个数在A中所有对应位置中upper_bound一下就好了
复杂度 $O(nlogn)$

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
const int M=1000005;
int n,m;
vector <int> pos[M];

void solve() {
    int len;
    scanf("%d",&len);
    int to=0;
    bool chk=1;
    for (int i=1;i<=len;++i) {
        int x;
        scanf("%d",&x);
        int id=upper_bound(pos[x].begin(),pos[x].end(),to)-pos[x].begin();
        if (id==pos[x].size()) {
            chk=0;
        } else to=pos[x][id];
    }
    printf("%s\n",(chk?"TAK":"NIE"));
}

int main() {
    cin>>n;
    for (int i=1;i<=n;++i) {
        int x;
        scanf("%d",&x);
        pos[x].push_back(i);
    }
    cin>>m;
    while (m--) solve();
    return 0;
}