动态规划 / 树状DP----G将军敢死队

G将军敢死队

题目

G将军有一支训练有素的军队，这个军队除开G将军外，每名士兵都有一个直接上级（可能是其他士兵，也可能是G将军）。现在G将军将接受一个特别的任务，需要派遣一部分士兵（至少一个）组成一个敢死队，为了增加敢死队队员的独立性，要求如果一名士兵在敢死队中，他的直接上级不能在敢死队中。请问，G将军有多少种派出敢死队的方法。注意，G将军也可以作为一个士兵进入敢死队。输入格式：输入的第一行包含一个整数n，表示包括G将军在内的军队的人数。军队的士兵从1至n编号，G将军编号为1。接下来n-1个数，分别表示编号为2, 3, …, n的士兵的直接上级编号，编号i的士兵的直接上级的编号小于i。输出格式：输出一个整数，表示派出敢死队的方案数。由于数目可能很大，你只需要输出这个数除10007的余数即可。
样例输入1
3
1 1
样例输出1
4
样例说明
这四种方式分别是：1. 选1；2. 选2；3. 选3；4. 选2, 3。
样例输入2
7
1 1 2 2 3 3
样例输出2
40

解答

#include<stdio.h>
#include<vector>
using namespace std;
vector<int> io[100001];
int dp[100001][2];

void DP(int x){
	int child;
	dp[x][0]=dp[x][1]=1;
    for(int i=0;i<io[x].size();i++){
    	child=io[x][i];
    	DP(child);
    	dp[x][0]=dp[x][0]*(dp[child][0]+dp[child][1])%10007;
    	dp[x][1]=dp[x][1]*dp[child][0]%10007;
    }
}
main(){
	int m,x;
	scanf("%d",&m);
	for(int i=1;i<=m;i++) io[i].clear();
	for(int i=2;i<=m;i++){
		scanf("%d",&x);
		io[x].push_back(i);
	}
	DP(1);
	printf("%d",(dp[1][0]+dp[1][1]-1)%10007);
}