洛谷 P1955[NOI2015]程序自动分析【离散化+并查集】

题目描述

在实现程序自动分析的过程中，常常需要判定一些约束条件是否能被同时满足。
考虑一个约束满足问题的简化版本：假设x1,x2,x3…代表程序中出现的变量，给定n个形如xi=xj或xi≠xj的变量相等/不等的约束条件，请判定是否可以分别为每一个变量赋予恰当的值，使得上述所有约束条件同时被满足。例如，一个问题中的约束条件为：x1=x2,x2=x3,x3=x4,x4≠x1，这些约束条件显然是不可能同时被满足的，因此这个问题应判定为不可被满足。

现在给出一些约束满足问题，请分别对它们进行判定。

输入格式：

从文件prog.in中读入数据。

输入文件的第1行包含1个正整数t，表示需要判定的问题个数。注意这些问题之间是相互独立的。

对于每个问题，包含若干行：
第1行包含1个正整数n，表示该问题中需要被满足的约束条件个数。接下来n行，每行包括3个整数i,j,e，描述1个相等/不等的约束条件，相邻整数之间用单个空格隔开。若e=1，则该约束条件为xi=xj；若�e=0，则该约束条件为xi≠xj；

输出格式：

输出到文件 prog.out 中。

输出文件包括t行。
输出文件的第 k行输出一个字符串“ YES” 或者“ NO”（不包含引号，字母全部大写），“ YES” 表示输入中的第k个问题判定为可以被满足，“ NO” 表示不可被满足。

输入样例

2
2
1 2 1
1 2 0
2
1 2 1
2 1 1

输出样例

NO
YES
说明

【样例解释1】

在第一个问题中，约束条件为：x1=x2,x1≠x2。这两个约束条件互相矛盾，因此不可被同时满足。

在第二个问题中，约束条件为：x1=x2,x1=x2。这两个约束条件是等价的，可以被同时满足。

【样例说明2】

在第一个问题中，约束条件有三个：x1=x2,x2=x3,x3=x1。只需赋值使得x1=x1=x1，即可同时满足所有的约束条件。

在第二个问题中，约束条件有四个：x1=x2,x2=x3,x3=x4,x4≠x1。由前三个约束条件可以推出x1=x2=x3=x4，然而最后一个约束条件却要求x1≠x4，因此不可被满足。

【数据范围】

【时限2s，内存512M】

---

题目分析：

省选水题 然而我还是交了四次才A

不考虑i、j范围情况下，思路不难想
读入所有操作
对所有xi==xj合并到一个集合
再离线查询所有xi！=xj
若发现xi、xj属于同一个集合
则直接输出No

而对于i、j超巨大的范围
也很容易想到离散化
所以这题就完美解决啦

---

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#include<vector>
using namespace std;

int read()
{
    int f=1,x=0;
    char ss=getchar();
    while(ss<'0'||ss>'9'){if(ss=='-')f=-1;ss=getchar();}
    while(ss>='0'&&ss<='9'){x=x*10+ss-'0';ss=getchar();}
    return x*f;
}

int t;
int n;
int fa[500010];
int pos[500010];
int cnt,tot;
struct node{int x,y,e;}a[500010];
int num[500010];
bool judge=true;

int find(int x)
{
    int a=x;
    while(x!=fa[x]) x=fa[x];
    while(a!=fa[a])
    {
        int temp=a;
        a=fa[a];
        fa[temp]=x;
    }
    return x;
}

int query(int x)
{
    return lower_bound(pos+1,pos+cnt+1,x)-pos;
}

bool cmp(node a,node b)
{
    return a.e>b.e;
}

void init()
{
    cnt=tot=0; judge=true;
    memset(pos,0,sizeof(pos));
    memset(num,0,sizeof(num));
}

int main()
{
    t=read();
    while(t--)
    {
        n=read(); 
        init();

        for(int i=1;i<=n;i++)
        {
            int x=read(),y=read(),e=read();
            a[i].x=x; a[i].y=y; a[i].e=e;//记录所有关系
            num[++tot]=x; num[++tot]=y;//记录所有出现过的数
        }

        sort(num+1,num+1+tot);//将所有数排序后离散化
        for(int i=1;i<=tot;i++)
        if(i==1||num[i]!=num[i-1])
        pos[++cnt]=num[i];

        sort(a+1,a+1+n,cmp);//把所有1操作放在前面进行
        for(int i=1;i<=cnt;i++) fa[i]=i;
        for(int i=1;i<=n;i++)
        {
            int tx=query(a[i].x),ty=query(a[i].y);
            //查询该数值对应的离散化编号
            int fu=find(tx),fv=find(ty);

            if(a[i].e==1&&fu!=fv)  
            fa[fu]=fv;//合并

            else if(a[i].e==0&&fu==fv)
            {judge=false;break;}//判断0操作是否与前面冲突
        }
        if(judge) printf("YES\n");
        else printf("NO\n");
    }
    return 0;
}