CF543D Road Improvement （换根dp，前后缀积）

link

• 我们很容易推出dp的式子 dp[u]为所有的（dp[v]+1)的乘积
• 因为 如果我们修这条路 贡献为 dp[v] 不修的话 为 1
• 我们需要统计每个点的答案 所以我们直接换根 就行
• 维护出 前后缀积 每次就可以o1 的获得 父亲的贡献
• 然后直接做dp就行 代码有很多细节

#include <bits/stdc++.h>
/*#include <iostream>
#include <climits>
#include <cstring>
#include <algorithm>
#include <queue>
#include <vector>
#include <bitset>
*/
using namespace  std;
#define  int long long
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef vector<int> vi;
#define fi first
#define se second
#define pb  push_back
#define inf 1ll<<62
#define db double
#define endl "\n"
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define de_bug(x) cerr << #x << "=" << x << endl
#define all(a) a.begin(),a.end()
#define IOS   std::ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define  fer(i,a,b)  for(int i=a;i<=b;i++)
#define  der(i,a,b)  for(int i=a;i>=b;i--)
const int mod = 1e9 + 7;
const int N = 1e6 + 10;
int n, m , k;
vi g[N];
vi suf[N];
vi pre[N];
int dp[N];
int ans[N];
//树形dp 前后积模型
void dfs1(int u, int fa) {
	dp[u] = 1;
	for(auto v : g[u]) {
		if(v == fa)continue;
		dfs1(v, u);
		(dp[u] *= (dp[v] + 1)) %= mod ;
	}
}
void dfs2(int u, int fa) {
	ans[u] = 1;
	for(auto v : g[u]) {
		(ans[u] *= (dp[v] + 1)) %= mod;
		if(v != fa)pre[u].push_back(dp[v] + 1), suf[u].push_back(dp[v] + 1);
	}
	for(int i = 1; i < (int)pre[u].size(); i++) pre[u][i] = pre[u][i] * pre[u][i - 1] % mod;
	for(int i = (int)suf[u].size() - 2; i >= 0; i--) suf[u][i] = suf[u][i] * suf[u][i + 1] % mod;
	int cnt = 0;
	for(auto v : g[u]) {
		if(v == fa)continue;
		dp[u] = (fa ? dp[fa] + 1 : 1);
		if(cnt > 0) dp[u] = dp[u] * pre[u][cnt - 1] % mod;
		if(cnt < suf[u].size() - 1) dp[u] = dp[u] * suf[u][cnt + 1] % mod;
		dfs2(v, u);
		cnt++;
	}

}
void solve() {
	cin >> n;
	fer(i, 2, n ) {
		int x;
		cin >> x;
		g[x].push_back(i);
		g[i].push_back(x);
	}
	dfs1(1, 0);
	dfs2(1, 0);
	fer(i, 1, n) {
		cout << ans[i] << " ";
	}
	cout << endl;
}
signed main() {
	IOS;
	int _ = 1;
	//cin>>_;
	while( _-- )
		solve();
	return 0;
}