洛谷 P1955 [NOI2015] 程序自动分析(并查集,离散化)

传送门


解题思路

很简单的一道并查集,但是我自己却没做出来,关键点没想到,害,还是太菜了……
直接放重点:

  • 先构建相等关系,然后判断不相等关系成不成立,所以分开操作。
  • 离散化。

AC代码

#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<set>
#include<map>
#include<vector>
#include<iomanip>
#include<ctime>
#include<stack>
using namespace std;
const int maxn=2e6+5;
int t,n,a[maxn],fa[maxn],cnt,newn;
struct node{
	int tp,x,y;
}q[maxn];
int find(int x){
	if(fa[x]==x) return x;
	return fa[x]=find(fa[x]);
}
inline void merge(int x,int y){
	fa[find(x)]=find(y);
}
bool cmp(node a,node b){
	return a.tp>b.tp;
}
int main()
{
    ios::sync_with_stdio(false);
    cin>>t;
    while(t--){
    	memset(a,0,sizeof(a));
    	cnt=0;
    	cin>>n;
    	for(int i=1;i<=n;i++) cin>>q[i].x>>q[i].y>>q[i].tp,a[++cnt]=q[i].x,a[++cnt]=q[i].y;
    	sort(a+1,a+cnt+1);
    	newn=unique(a+1,a+cnt+1)-a-1;
    	for(int i=0;i<=newn;i++) fa[i]=i;
    	for(int i=1;i<=n;i++){
    		q[i].x=lower_bound(a+1,a+newn+1,q[i].x)-a;
    		q[i].y=lower_bound(a+1,a+newn+1,q[i].y)-a;
		}
		sort(q+1,q+n+1,cmp);
		for(int i=1;i<=n;i++){
			if(q[i].tp==1) merge(q[i].x,q[i].y);
			else{
				if(find(q[i].x)==find(q[i].y)){
					cout<<"NO"<<endl;
					break;
				}
			}
			if(i==n) cout<<"YES"<<endl;
		}
		
	}
    return 0;
}

//NOI2015 D1T1

上一篇:计算机视觉-全景拼接


下一篇:数据挖掘(六)——模型优化与评估方法