牛客 26C 手铐 (缩环, 树形dp)

先缩环建树, 对于树上个环$x,y$, 假设$x,y$路径上有$cnt$个环(不包括$x,y$), 贡献就为$2^{cnt}$.

这题卡常挺严重的, 刚开始用并查集合并竟然T了.

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <queue>
#define REP(i,a,n) for(int i=a;i<=n;++i)
#define pb push_back
using namespace std;
typedef long long ll;

const int N = 1e6+10, P = 0x0125e591;
int n,m,ans,cnt,dep[N],fa[N];
int s[N],vis[N],dp[N];
vector<int> g[N],gg[N];

void get(int x, int y) {
    if (dep[x]<dep[y]) return;
	s[y]=++cnt,vis[cnt]=1;
    for (; x!=y; x=fa[x]) s[x]=cnt;
}
void dfs(int x, int f) {
    fa[x]=f,dep[x]=dep[f]+1;
    for (int y:g[x]) if (y!=f) {
        if (dep[y]) get(x,y);
        else dfs(y,x);
    }
}
void dfs2(int x, int f) {
	int t = vis[x]+1;
	dp[x] = vis[x];
	for (int y:gg[x]) if (y!=f) {
		dfs2(y,x);
		ans = (ans+(ll)dp[x]*dp[y])%P;
		dp[x] = (dp[x]+t*dp[y])%P;
	}
}

int main() {
	scanf("%d%d",&n,&m);
	REP(i,1,m) {
		int u, v;
		scanf("%d%d",&u,&v);
		g[u].pb(v),g[v].pb(u);
	}
	dfs(1,0);
	REP(i,1,n) if (!s[i]) s[i]=++cnt;
	REP(i,1,n) for (int j:g[i]) {
		int u=s[i],v=s[j];
		if (u!=v) gg[u].pb(v);
	}
	dfs2(1,0);
	if (ans<0) ans += P;
	printf("%d\n", ans);
}