Dmy_树上联通块

#include <bits/stdc++.h>
#define rep(i, a, b) for(int i = (a); i <= (b); i ++ )
using namespace std;
using LL = long long;
template <typename T> void chkmax(T &x, T y) { x = max(x, y); }
template <typename T> void chkmin(T &x, T y) { x = min(x, y); }

constexpr int N = 1e5 + 10;

int n, MOD;
LL dp[N], pre[N], suf[N], pd[N], ans[N];
vector<int> edge[N];

void dfs(int u) {
	dp[u] = 1;
	for(int &v: edge[u]) {
		dfs(v);
		dp[u] = dp[u] * (dp[v] + 1) % MOD;
	}
}

void dfs2(int u) {
	int m = edge[u].size();
	//pre[i] 数组是处理 u 的子树中前 i 个子树所能产生的连通块的总数目
	pre[0] = 1;
	for(int i = 0; i < m; i ++ ) {
		pre[i + 1] = pre[i] * (dp[edge[u][i]] + 1) % MOD;
	}
	//suf[i]数组是处理 u 的子树中后 n - i  -> n 的所有子树所能产生的连通块的总数目
	suf[m] = 1;
	for(int i = m - 1; i >= 0; i -- ) {
		suf[i] = suf[i + 1] * (dp[edge[u][i]] + 1) % MOD;	
	}
	//pd数组是 u 的父节点所产生的 连通块的总数目
	//利用乘法原理可以得出结果
	for(int i = 0; i < m; i ++ ) {
		int v = edge[u][i];
		pd[v] = (pd[u] + 1) * pre[i] % MOD * suf[i + 1] % MOD;
		ans[v] = (pd[v] + 1) * dp[v] % MOD;
	}

	for(int &v: edge[u]) {
		dfs2(v);
	}

}

int main() {
	#ifndef ONLINE_JUDGE
 		freopen("1.txt", "r", stdin);
 		freopen("2.txt", "w", stdout);
 	#endif
	
	scanf("%d%d", &n, &MOD);

	for(int i = 2; i <= n; i ++ ) {
		int x; scanf("%d", &x);
		edge[x].push_back(i);
	}

	dfs(1);
	dfs2(1);

	ans[1] = dp[1];

	for(int i = 1; i <= n; i ++ ) {
		printf("%lld\n", ans[i]);
	}

    return 0;

}