[BZOJ2083][Poi2010]Intelligence test（sort+二分）

题目描述

传送门

题解

这个题显然要是能把数都读进来就是可以$O(nlogn)$搞的。
首先将原数列的数按照权值第一关键字，位置第二关键字排序，确定每一个值所在区间的左右端点。
每读入一个数，贪心的思想，二分查找在这个值区间内位置大于上一个数的位置且最靠左的点。如果无法找到则无解。

代码

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
using namespace std;
#define N 1000005

int n,m,x,last;
struct hp{int val,loc;}a[N];
int l[N],r[N];
bool flag;

int read()
{
    int x=0; char ch=getchar();
    while (ch<'0'||ch>'9') ch=getchar();
    while (ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x;
}
int cmp(hp a,hp b)
{
    return a.val<b.val||(a.val==b.val&&a.loc<b.loc);
}
int find(int l,int r,int id)
{
    int mid,ans=-1;
    while (l<=r)
    {
        mid=(l+r)>>1;
        if (a[mid].loc>id) ans=a[mid].loc,r=mid-1;
        else l=mid+1;
    }
    return ans;
}
int main()
{
    m=read();
    for (int i=1;i<=m;++i) a[i].val=read(),a[i].loc=i;
    sort(a+1,a+m+1,cmp);l[a[1].val]=1;r[a[m].val]=m;
    for (int i=2;i<=m;++i)
        if (a[i].val!=a[i-1].val) r[a[i-1].val]=i-1,l[a[i].val]=i;
    n=read();
    for (int i=1;i<=n;++i)
    {
        m=read();flag=true;last=0;
        for (int j=1;j<=m;++j)
        {
            x=read();
            if (last!=-1) last=find(l[x],r[x],last);
            if (last==-1) flag=false;
        }
        if (flag) puts("TAK");
        else puts("NIE");
    }
}