【jzoj4822】【完美标号】【2-set】

题目大意

给定M个二元组(A_i, B_i),求X_1, …, X_N满足:对于任意(A_i, B_i),有|X_{A_i} - X_{B_i}| = 1成立。

解题思路

我们发现其实这是二分图判定,我们只要用0,1染色即可,直接dfs。

code

#include
#include
#include
#include
#include
#include
#define LL long long
#define max(a,b) ((a>b)?a:b)
#define min(a,b) ((a>b)?b:a)
#define fo(i,j,k) for(int i=j;i<=k;i++)
#define fd(i,j,k) for(int i=j;i>=k;i--)
using namespace std;
int const inf=2147483647;
int const maxn=10000,maxm=100000;
int n,m,gra,ok,to[maxm*2+10],next[maxm*2+10],begin[maxn+10],col[maxn+10];
void insert(int u,int v){
    to[++gra]=v;
    next[gra]=begin[u];
    begin[u]=gra;
}
void dfs(int now){
    for(int i=begin[now];i;i=next[i])
        if(!col[to[i]]){
            col[to[i]]=3-col[now];
            dfs(to[i]);
            if(!ok)return;
        }else if(col[to[i]]+col[now]!=3){
            ok=0;
            return;
        }
}
int main(){
    freopen("perfect.in","r",stdin);
    freopen("perfect.out","w",stdout);
    scanf("%d%d",&n,&m);
    fo(i,1,m){
        int u,v;scanf("%d%d",&u,&v);
        insert(u,v);insert(v,u);
    }
    ok=1;
    fo(i,1,n)
        if(!col[i]){
            col[i]=1;
            dfs(i);
        }
    if(ok){
        printf("YES\n");
        fo(i,1,n)printf("%d ",col[i]);
    }else printf("NO\n");
    return 0;
}

你可能感兴趣的:(jzoj,搜索)