BZOJ 4195 程式自動分析 並查集
Description
在實現程式自動分析的過程中,常常需要判定一些約束條件是否能被同時滿足。
考慮一個約束滿足問題的簡化版本:假設x1,x2,x3,…代表程式中出現的變數,給定n個形如xi=xj或xi≠xj的變數相等/不等的約束條件,請判定是否可以分別為每一個變數賦予恰當的值,使得上述所有約束條件同時被滿足。例如,一個問題中的約束條件為:x1=x2,x2=x3,x3=x4,x1≠x4,這些約束條件顯然是不可能同時被滿足的,因此這個問題應判定為不可被滿足。
現在給出一些約束滿足問題,請分別對它們進行判定。
Input
輸入檔案的第1行包含1個正整數t,表示需要判定的問題個數。注意這些問題之間是相互獨立的。
對於每個問題,包含若干行:
第1行包含1個正整數n,表示該問題中需要被滿足的約束條件個數。
接下來n行,每行包括3個整數i,j,e,描述1個相等/不等的約束條件,相鄰整數之間用單個空格隔開。若e=1,則該約束條件為xi=xj;若e=0,則該約束條件為xi≠xj。
Output
輸出檔案包括t行。
輸出檔案的第k行輸出一個字串“YES”或者“NO”(不包含引號,字母全部大寫),“YES”表示輸入中的第k個問題判定為可以被滿足,“NO”表示不可被滿足。
Sample Input
2
2
1 2 1
1 2 0
2
1 2 1
2 1 1
Sample Output
NO
YES
HINT
在第一個問題中,約束條件為:x1=x2,x1≠x2。這兩個約束條件互相矛盾,因此不可被同時滿足。
在第二個問題中,約束條件為:x1=x2,x2=x1。這兩個約束條件是等價的,可以被同時滿足。
1≤n≤1000000
1≤i,j≤1000000000
思路:並查集+離散化
1.每次二分查詢離散化的下標
#include <cstdio> #include <algorithm> #include <iostream> #include <cstring> #include <cmath> #include <set> #include <string> using namespace std; const int MAXN = 2e6+5; int parent[MAXN]; int mp[MAXN]; int x[MAXN]; int y[MAXN]; int z[MAXN]; int root(int x) { if(x==parent[x]) return x; return parent[x]=root(parent[x]); } void make(int x,int y) { int fx=root(x); int fy=root(y); parent[fx]=fy; } int main() { int t; scanf("%d",&t); while(t--){ int n,num=1; scanf("%d",&n); for(int i=1;i<=2*n;i++) parent[i]=i; memset(mp,0,sizeof(mp)); for(int i=1;i<=n;i++) { scanf("%d%d%d",&x[i],&y[i],&z[i]); mp[num++]=x[i]; mp[num++]=y[i]; } sort(mp+1,mp+num); int flag=0; for(int i=1;i<=n;i++) if(z[i]==1){ int x1=lower_bound(mp+1,mp+num,x[i])-mp; int y1=lower_bound(mp+1,mp+num,y[i])-mp; //printf("%d %d\n",x1,y1); int fx=root(x1); int fy=root(y1); if(fx!=fy) make(fx,fy); } for(int i=1;i<=n;i++) if(z[i]==0){ int x1=lower_bound(mp+1,mp+num,x[i])-mp; int y1=lower_bound(mp+1,mp+num,y[i])-mp; // printf("%d %d\n",x1,y1); int fx=root(x1); int fy=root(y1); if(fx==fy){ flag=1; break; } } if(!flag) printf("YES\n"); else printf("NO\n"); } return 0; }
2.利用結構體存貯下標順序後再離散化
#include <cstdio>
#include <algorithm>
#include <iostream>
#include <cstring>
#include <cmath>
#include <set>
#include <string>
using namespace std;
const int MAXN = 2e6+5;
int parent[MAXN];
struct node
{
int id;
int v;
}a[MAXN];
int b[MAXN];
int z[MAXN];
bool cmp(node a,node b)
{
return a.v<b.v;
}
int root(int x)
{
if(x==parent[x])
return x;
return parent[x]=root(parent[x]);
}
void make(int x,int y)
{
int fx=root(x);
int fy=root(y);
parent[fx]=fy;
}
int main()
{
int t;
scanf("%d",&t);
while(t--){
int n,num=1,x,y;
scanf("%d",&n);
for(int i=1;i<=2*n;i++)
parent[i]=i;
memset(b,0,sizeof(b));
for(int i=1;i<=n;i++)
{
scanf("%d%d%d",&x,&y,&z[i]);
a[num].v=x;
a[num+1].v=y;
a[num].id=num;
a[num+1].id=num+1;
num=num+2;
}
sort(a+1,a+1+2*n,cmp);
b[a[1].id]=1;
for(int i=2;i<=2*n;i++){
if(a[i].v!=a[i-1].v)
b[a[i].id]=i;
else
b[a[i].id]=b[a[i-1].id];
}
// for(int i=1;i<=2*n;i++)
// printf("%d\n",b[i]);
int flag=0;
for(int i=1;i<=n;i++)
if(z[i]==1)
make(b[2*i-1],b[2*i]);
for(int i=1;i<=n;i++)
if(z[i]==0){
int fx=root(b[2*i-1]);
int fy=root(b[2*i]);
if(fx==fy){
flag=1;
break;
}
}
if(!flag)
printf("YES\n");
else
printf("NO\n");
}
return 0;
}