[边权转点权] 边的染色 美团2018年CodeM大赛-复赛

题面

link
  给定一张无向图，有些边已经被标记成 $0,1$，我们需要给未标记的边赋值 $0，1$，使得无向图中任意一个环中所有边的异或和为0。问有几种方案？

分析

  先从简单的入手，给定一个连通图，把里面的边赋值 $0，1$，使得任意一个环，里面所有边的边权异或值为0，求方案数。
  我们可以不直接考虑边如何赋值，可以考虑点。在一个环中，每个点都会贡献两条边，若是我们定义边权为两个顶点的异或和，那么不管点权是多少，在环中一个边的权值可以拆成两个点权值的异或，一个顶点都会被异或两次，那么边权异或和肯定为0。
  那么任意一种点权取值就对应了一种边权取值，但我们要注意若是两种方案的点权取值完全相反，即 一种方案的 $0$ 在另一种方案中全是 $1$， $1$ 全是 $0$，那么对应的边权取值是一样的。所以一个有 $k$ 个顶点的连通图，方案数为 $2^{k-1}$ 种。
  那么对于一个无向图来说，只要拆成各个连通图考虑然后相乘就可以了。
  
  但有的边权已经确定了怎么办？若一条边的边权已经确定，那么其中一个顶点值确定，另一个顶点值也相应确定了，即不能任意取值了。我们可以推得一个边权确定的连通图(顶点都由确定边权的边连接)，只要其中一个顶点确定，其他顶点也会被确定，若这个图有 $k$ 个顶点，即答案需要除掉 $2^{k-1}$。这里需要注意的是，可能有一个环的边都已经确定但是异或和不为0，这时候直接方案数就为0。
  所以我们边权转点权之后，需要两遍 dfs，分别求出相应的连通图。
  

#include <bits/stdc++.h>
#define pb push_back

using namespace std;

typedef long long ll;
typedef pair<int, int> P;
const int maxn = 1e5 + 10;
const int INF = 0x3f3f3f3f;
const ll mod = 998244353;

ll qpow(ll x, ll n)
{
    ll t = 1;
    for (; n; n >>= 1, x = x * x % mod)
        if (n & 1)
            t = t * x % mod;
    return t;
}

struct edge
{
    int to, next, w;
}e[maxn*2];

int head[maxn], tol;

void add_edge(int x, int y, int w)
{
    e[++tol].to = y;
    e[tol].w = w;
    e[tol].next = head[x];
    head[x] = tol;
}

int n, m, vis[maxn], cur, mark[maxn], ans;

bool dfs1(int x)
{
    vis[x] = 1;
    cur++;
    for(int i = head[x]; i; i = e[i].next)
    {
        int to = e[i].to, w = e[i].w;
        if(w == -1)  continue;
        if(vis[to])
        {
            if((mark[x] ^ mark[to]) != w) return false;
        }
        else
        {
            mark[to] = mark[x] ^ w;
            if(!dfs1(to))  return false;
        }
    }
    return true;
}

void dfs2(int x)
{
    vis[x] = 1;
    cur++;
    for(int i = head[x]; i; i = e[i].next)
    {
        int to = e[i].to;
        if(vis[to])  continue;
        dfs2(to);
    }
}

int main()
{
    scanf("%d %d", &n, &m);
    for(int i = 1; i <= m; i++)
    {
        int x, y, z;
        scanf("%d %d %d", &x, &y, &z);
        add_edge(x, y, z);
        add_edge(y, x, z);
    }

    for(int i = 1; i <= n; i++)        //求确定边权的连通图
    {
        if(!dfs1(i))
        {
            printf("%d\n", 0);
            return 0;
        }
        ans -= cur - 1;
        cur = 0;
    }

    fill(vis + 1, vis + 1 + n, 0);
    for(int i = 1; i <= n; i++)         //求连通图
    {
        dfs2(i);
        ans += cur - 1;
        cur = 0;
    }

    printf("%lld\n", qpow(2, ans));
}