BZOJ2083: [Poi2010]Intelligence test

2083: [Poi2010]Intelligence test

Time Limit: 10 Sec  Memory Limit: 259 MB
Submit: 241  Solved: 96
[Submit][Status]

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

HINT

Source

by poi

题解：
被这道题坑了。。。
刚开始还是没思路，后来前天忽然想到是不是可以记录一个数的后继出现位置然后用mx表示当前最右位置，然后每次碰到x，都让x next 到>mx 的第一个位置
这样的复杂度是多少呢？感觉是O（n）的，最多不会next超过n次，然后就开开心心打代码。
WA了几发之后改了改，然后又T了。。。
不知道原因。。。
代码：

#include<cstdio>

#include<cstdlib>

#include<cmath>

#include<cstring>

#include<algorithm>

#include<iostream>

#include<vector>

#include<map>

#include<set>

#include<queue>

#include<string>

#define inf 1000000000

#define maxn 1000000+5

#define maxm 500+100

#define eps 1e-10

#define ll long long

#define pa pair<int,int>

#define for0(i,n) for(int i=0;i<=(n);i++)

#define for1(i,n) for(int i=1;i<=(n);i++)

#define for2(i,x,y) for(int i=(x);i<=(y);i++)

#define for3(i,x,y) for(int i=(x);i>=(y);i--)

#define mod 1000000007

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=10*x+ch-'0';ch=getchar();}

    return x*f;

}
int n,a[maxn],b[maxn],cur[maxn],next[maxn],head[maxn];

int main()

{

    freopen("input.txt","r",stdin);

    freopen("output.txt","w",stdout);

    n=read();
    for1(i,n)a[i]=read();
    for3(i,n,1)
    {
        next[i]=head[a[i]];
        head[a[i]]=i;
    }
    memset(cur,-1,sizeof(cur));
    int cs=read();
    while(cs--)
    {
        int i,x,mx=1,m=read();
        for(i=1;i<=m;i++)b[i]=read();
        for(i=1;i<=m;i++)
        {
            x=b[i];
            if(cur[x]==-1)cur[x]=head[x];else cur[x]=next[cur[x]];
            while(cur[x]<mx&&cur[x])cur[x]=next[cur[x]];
            //cout<<i<<' '<<mx<<' '<<x<<' '<<cur[x]<<endl;
            if(cur[x]<=0)break;
            mx=cur[x];
        }
        if(i<=m)printf("NIE
");else printf("TAK
");
        for(i=1;i<=m;i++)cur[b[i]]=-1;
    }

    return 0;

}

二分的思路其实差不多，但多一个log，但是能A。汗！

代码：

#include<cstdio>

#include<cstdlib>

#include<cmath>

#include<cstring>

#include<algorithm>

#include<iostream>

#include<vector>

#include<map>

#include<set>

#include<queue>

#include<string>

#define inf 1000000000

#define maxn 1000000+5

#define maxm 500+100

#define eps 1e-10

#define ll long long

#define pa pair<int,int>

#define for0(i,n) for(int i=0;i<=(n);i++)

#define for1(i,n) for(int i=1;i<=(n);i++)

#define for2(i,x,y) for(int i=(x);i<=(y);i++)

#define for3(i,x,y) for(int i=(x);i>=(y);i--)

#define mod 1000000007
#define it vector<int>::iterator

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=10*x+ch-'0';ch=getchar();}

    return x*f;

}
int n,a[maxn];
vector<int>v[maxn];

int main()

{

    freopen("input.txt","r",stdin);

    freopen("output.txt","w",stdout);

    n=read();
    for1(i,n)v[read()].push_back(i);
    int cs=read();
    while(cs--)
    {
        int m=read(),mx=0;bool flag=1;
        for1(i,m)
        {
            int x=read();
            if(flag)
            {
                it y=upper_bound(v[x].begin(),v[x].end(),mx);
                if(y==v[x].end())flag=0;else mx=*y;
            }
        }
        if(flag)puts("TAK");else puts("NIE");
    }

    return 0;

}