我的思路貌似和大家不一样哈……
题面描述:
很简单:
给出许多相等和不相等的条件,判断是否矛盾。
思路:
其实这道题连蓝题都不到……
使用并查集统计不同的关系,再枚举相同的,看是否矛盾即可。
发现i,j极大,冷静,不要写哈希!
记住,先看一下可不可以使用map。
发现可以,开两个map,一个用于存储并查集的数组,另一个则存储数组中有没有出现过这个数,再编写一个函数,用于返回这个数组上的数的引用,这样如果下面要改的话很方便。
这个函数:(unordered_map是啥看我的文章)
long long n,nd,p,di[1000005],dj[1000005];
unordered_map<long long,long long> idm;
unordered_map<long long,bool> idb;
long long &id(long long x){
if(idb[x]) return idm[x];//如果已经有了,就直接返回数组中的数
else {//否则返回x
idb[x]=1,idm[x]=x;
return idm[x];
}
}
下面贴出全篇代码:
#include<bits/stdc++.h>
using namespace std;
long long n,nd,p,di[1000005],dj[1000005];
unordered_map<long long,long long> idm;//存储并查集数组
unordered_map<long long,bool> idb;//存储是否出现过
long long &id(long long x){ //返回的是引用
if(idb[x]) return idm[x];
else {
idb[x]=1,idm[x]=x;
return idm[x];
}
}
long long find(long long x){
if(id(x)==x) return x;
return id(x)=find(id(x));
}
void unite(long long x,long long y){
long long xi=find(x),yi=find(y);
id(xi)=yi;//因为是引用,所以可以直接这样写
}
int main(){
ios::sync_with_stdio(false);//读入优化
cin>>p;
for(int np=0;np<p;np++){
idm.clear();//清空map
idb.clear();
nd=0;
bool flag=0;
cin>>n;
int i,j,e;
for(int a=0;a<n;a++){
cin>>i>>j>>e;
if(e==0){//如果两者不相同,把它们存起来
di[nd]=i,dj[nd]=j;
nd++;
}
else unite(i,j);//否则将它们存入并查集
}
for(int a=0;a<nd;a++){
if(find(di[a])==find(dj[a])){//如果两者本应相等,却不相等
cout<<"NO\n";
flag=1;
break;
}
}
if(!flag)cout<<"YES\n";
}
return 0;
}
再说一下,这个代码不开优化是不过的,必须开O2优化才能过。
最后一次更新于2019-08-06
0 条评论