题目描述 独立集是指两两不相邻的节点构成的集合。 求一棵树中,独立集的个数。(空集也算独立集) 输入 第一行一个整数n,表示树中节点的个数。
第二行n-1个数,表示每个节点父亲节点的编号。父节点的编号一定小于该节点。 输出
输出一个整数,表示独立集的个数,方案数较多,对1000000007取模。
这是道由2012年NOIP初赛问题求解第二道改编过来的题目
可以用树形dp的思想求解:
dp[i][0]=dp[i][0]*(dp[j][1]+dp[j][0])
dp[i][1]=dp[i][1]*dp[j][0]
j为i的儿子,只要将信息收集上来就好了
由于父亲比儿子小,所以可以不dfs
#include<cstdio>
#define M 100005
#define Mod 1000000007
int fa[M],dp[2][M];
int main(){
int n,x;
scanf("%d",&n);
for(int i=2;i<=n;i++){
scanf("%d",&x);
fa[i]=x;
dp[1][i]=dp[0][i]=1;
}
dp[1][1]=dp[0][1]=1;
for(int i=n;i>=2;i--){
int y=fa[i];
dp[0][y]=1LL*dp[0][y]*(dp[0][i]+dp[1][i])%Mod;
dp[1][y]=1LL*dp[1][y]*dp[0][i]%Mod;
}
printf("%d\n",1LL*(dp[1][1]+dp[0][1])%Mod);
return 0;
}