237. 程序自动分析并查集

Posted 2021-10-27 幽殇默

https://www.acwing.com/problem/content/description/239/
这里的话，需要注意的是我们的范围特别的大，如果直接开数组会MLE。
所以这里我是直接用了哈希表来映射。然后我们保存状态，先处理全部的相等的条件，再处理所有的不等条件。
然后判断有没有矛盾点。
当然这里的方法有点极限卡时间。

#include<bits/stdc++.h>
using namespace std;
int t,n;
struct node {int a,b,c;}temp;
unordered_map<int,int>p;
bool cmp(node a,node b) {return a.c>b.c;}
int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
int main(void)
{
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        p.clear();
        vector<node>ve(n);
        for(int i=0;i<n;i++)
        {
            scanf("%d%d%d",&temp.a,&temp.b,&temp.c);
            if(!p[temp.a]) p[temp.a]=temp.a;
            if(!p[temp.b]) p[temp.b]=temp.b;
            ve[i]=temp;
        }
        sort(ve.begin(),ve.end(),cmp);//目的是让相等的在前面
        bool flag=true;
        for(int i=0;i<ve.size();i++)
        {
            int a=ve[i].a,b=ve[i].b,c=ve[i].c;
            if(c==1)
            {
                if(find(a)!=find(b)) p[find(a)]=find(b);
            }
            else
            {
                if(find(a)==find(b)) flag=false;//这俩在一个集合，冲突
            }
            if(!flag) break;
        }
        if(flag) puts("YES");
        else puts("NO");
    }
    return 0;
}

时间少的方法，用哈希表来完成离散化。

#include<bits/stdc++.h>
using namespace std;
unordered_map<int,int>hush;
const int N=1e5*2+10;
int t,n,m,p[N];
struct node{int a,b,c;};
int get(int x)//离散化
{
    if(hush.count(x)==0) hush[x]=++n;
    return hush[x];
}
int find(int x)
{
    if(x!=p[x]) p[x]=find(p[x]);
    return p[x];
}
int main(void)
{
    cin>>t;
    while(t--)
    {
        n=0,hush.clear();//初始化
        cin>>m;
        vector<node>ve(m);
        for(int i=0;i<m;i++)
        {
            int a,b,c; cin>>a>>b>>c;
            ve[i]={get(a),get(b),c};
        }
        for(int i=1;i<=n;i++) p[i]=i;
        for(int i=0;i<m;i++)
        {
            if(ve[i].c)
            {
                if(find(ve[i].a)!=find(ve[i].b)) p[find(ve[i].a)]=find(ve[i].b);
            }
        }
        bool flag=true;
        for(int i=0;i<m;i++)
        {
            if(!ve[i].c)
            {
                if(find(ve[i].a)==find(ve[i].b)) flag=false;
            }
            if(!flag) break;
        }
        if(flag) puts("YES");
        else puts("NO");
    }
    return 0;
}