欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

分组 ( dfs + 思维 )

程序员文章站 2022-06-08 16:30:01
...

题目链接
分组 ( dfs + 思维 )
解题报告:
将认识关系转化为图中的边。dfs这张图,对每一个没有被访问过的点,将它标记为源点的反色,回溯的时候统计每个点有多少同色相邻点,个数等于2时将其颜色转换。

#include<bits/stdc++.h>
#define LL long long
#define pii pair<LL,LL>
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
const int maxn=2e5+5;
const int inf=0x3f3f3f3f;
const LL MOD=1e9+7;

struct TMD{
    int to,next;
}edge[maxn<<1];
int n,m,head[maxn],cnt,knt,dfn[maxn],col[maxn];
void add(int from,int to){
    edge[++cnt]={to,head[from]};
    head[from]=cnt;
}
void dfs(int now){
    int t=0;dfn[now]=++knt;
    for(int i=head[now];i;i=edge[i].next){
        int to=edge[i].to;
        if(!dfn[to]){
            col[to]=col[now]^1;
            dfs(to);
        }
        t+=col[to]==col[now];
    }
    if(t>1) col[now]^=1;
}
int main() {
    int u,v;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    }
    for(int i=1;i<=n;i++){
        if(!dfn[i]) dfs(i);
    }
    for(int i=1;i<=n;i++) printf("%d%c",col[i]+1,i==n?'\n':' ');
    return 0;
}


相关标签: 树形结构 dfs