bzoj4195: [Noi2015]程序自动分析

bzoj4195: [Noi2015]程序自动分析
一次TLE：少个分号。
一次WA：e=0和e=1整个看反了。
这个故事告诉我们，不要被水题淹没不知所措。我们要认认真真地把它看完，哪怕它是道裸的并查集。

---

代码

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int T,i,t,n,lst,f[1000005];
bool e[1000005],ans;
struct dat{int val,num;}que[2000005];
bool cmp(dat a,dat b){return a.val<b.val;}
bool cp(dat a,dat b){return a.num<b.num;}
int fd(int x){return f[x]==x?x:f[x]=fd(f[x]);}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        t=ans=0;
        scanf("%d",&n);
        for(i=0;i<n;i++)
        que[i<<1].num=que[i<<1|1].num=i,
        scanf("%d%d%d",&que[i<<1].val,&que[i<<1|1].val,&e[i]);
        sort(que,que+(n<<1),cmp);
        lst=que[0].val,que[0].val=0;
        for(i=1;i<(n<<1);que[i++].val=t)
        if(que[i].val>lst)t++,lst=que[i].val;
        for(i=0;i<=t;f[i]=i++);
        sort(que,que+(n<<1),cp);
        for(i=0;i<n;i++)
        if(e[i])f[fd(que[i<<1].val)]=fd(que[i<<1|1].val);
        for(i=0;i<n;i++)
        if((!e[i])&&fd(que[i<<1].val)==fd(que[i<<1|1].val))
        {ans=1;break;}
        if(ans)printf("NO\n");
        else printf("YES\n");
    }
}