1. 程式人生 > >BZOJ 4195 程式自動分析 並查集

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;
}