codeforces 862B. Mahmoud and Ehab and the bipartiteness

最新推荐文章于 2021-07-20 14:39:08 发布

原创最新推荐文章于 2021-07-20 14:39:08 发布 · 271 阅读

0 ·

CC 4.0 BY-SA版权

ACM 专栏收录该内容

143 篇文章

订阅专栏

本文介绍了一种解决CodeForces平台B题的方法，通过构建二分图并使用染色法进行节点标记来求解。文章提供了完整的C++代码实现，并详细解释了如何通过深度优先搜索（DFS）为图中的每个节点分配颜色。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：

http://codeforces.com/contest/862/problem/B

题解：

二分图，染色法。

代码：

#include <map>
#include <cmath>
#include <vector>
#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define met(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
typedef long long ll;
const int maxn = 1e5+10;
vector<int> p[maxn];
int color[maxn],visited[maxn];

void dfs(int x,int c)
{
    visited[x]=1;
    color[x]=c;
    for(int i=0;i<p[x].size();i++)
    {
        int y=p[x][i];
        if(color[y]==0&&!visited[y])
        {
            dfs(y,-c);
        }
    }
}

int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        p[i].clear();
    int m=n-1;
    for(int i=0;i<m;i++)
    {
        int x,y;
        scanf("%d%d",&x,&y);
        p[x].push_back(y);
        p[y].push_back(x);
    }
    met(visited,0);
    met(color,0);
    dfs(1,1);
    ll ans1=0,ans2=0;
    for(int i=1;i<=n;i++)
    {
        if(color[i]==-1)
            ans1++;
        if(color[i]==1)
            ans2++;
    }
    ll ans=ans1*ans2-n+1;
    printf("%lld\n",ans);
}