Intelligence test(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。

样例输入

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

想法

• 从原序列中删除一些数 等价于 保存原序列中某些数的相对位置

算法

• 用f[i][j]表示i这个数出现第j次是在哪里
• 用一个vector来存储f数组
• 剩下的就是二分

代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <cstring>
#include <string>
#include <cmath>
#include <vector>
#define MAXN 1000005
using namespace std;
int n,x,m,len,b[MAXN],farest;
bool flag;
vector<int >q[MAXN];
inline int wbs(int x,int v)
{
    int l=0,r=q[x].size()-1,mid,pos=n+1;
    while(l<=r)
    {
        mid=(l+r)>>1;
        if(q[x][mid]>v)pos=q[x][mid],r=mid-1;
        else l=mid+1;
    }
    return pos;
}
int main()
{
    //freopen("test.in","r",stdin);
    //freopen("test.out","w",stdout);
    scanf("%d",&n);
    for (int i=1;i<=n;i++)
    {
        scanf("%d",&x);
        q[x].push_back(i);
    }
    scanf("%d",&m);
    for (int i=1;i<=m;i++)
    {
        farest=0;
        flag=true;
        scanf("%d",&len);
        for (int j=1;j<=len;j++)
            scanf("%d",&b[j]);
        for (int j=1;j<=len;j++)
        {
            farest=wbs(b[j],farest);
            if(farest>n){flag=false;break;}
        }
        if(flag==true)printf("TAK\n");
        else printf("NIE\n");
    }
    return 0;
}