G将军有一支训练有素的军队,这个军队除开G将军外,每名士兵都有一个直接上级(可能是其他士兵,也可能是G将军)。
现在G将军将接受一个特别的任务,需要派遣一部分士兵(至少一个)组成一个敢死队,
为了增加敢死队队员的独立性,要求如果一名士兵在敢死队中,
他的直接上级不能在敢死队中。
请问,G将军有多少种派出敢死队的方法。注意,G将军也可以作为一个士兵进入敢死队。
输入格式:
输入的第一行包含一个整数n,表示包括G将军在内的军队的人数。军队的士兵从1至n编号,G将军编号为1。
接下来n-1个数,分别表示编号为2, 3, …, n的士兵的直接上级编号,编号i的士兵的直接上级的编号小于i。
输出格式:
输出一个整数,表示派出敢死队的方案数。由于数目可能很大,你只需要输出这个数除10007的余数即可。
输入样例:
3
1 1
样例输出1
4
样例说明
这四种方式分别是:
- 选1;
- 选2;
- 选3;
- 选2, 3。
样例输入2
7
1 1 2 2 3 3
样例输出2
40
数据规模与约定
对于20%的数据,n ≤ 20;
对于40%的数据,n ≤ 100;
对于100%的数据,1 ≤ n ≤ 1007000。
资源约定:
峰值内存消耗 < 256M
CPU消耗 < 2000ms
请严格按要求输出,不要画蛇添足地打印类似:“请您输入…” 的多余内容。
所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
注意: main函数需要返回0
注意: 只使用ANSI C/ANSI C++ 标准,不要调用依赖于编译环境或操作系统的特殊函数。
注意: 所有依赖的函数必须明确地在源文件中 #include , 不能通过工程设置而省略常用头文件。
提交时,注意选择所期望的编译器类型。
思路:
当前节点A,当前节点的子节点B
- A被选中的次数=A被选中的次数*B不被选中的次数。//当前节点选中,其子节点不能被选。
- A不被选中的次数=A不被选中的次数*(B被选中的次数+B不被选中的次数)。//A节点不被选中,其子节点可以被选也可以不被选。
- 得到最终的总数=根节点被选中次数+根节点不被选中次数-1。//减一是减去全部都不选,队伍里没有一人的情况。
如果不能理解也可以把它当成概率题,
A节点被选中的概率=其子节点不被选中的概率的乘积。
A节点不被选中的概率=其子节点被选中概率+子节点不被选中概率的和的乘积。
代码:
#include<iostream>
#include<list>
#define MAX 100000
#define MOD 10007
using namespace std;
struct node{
int father;
list<int> son;
int son_count=0;
int chosed, abandoned;
}tree[MAX];
void DFS(int id)//通过递归,求每个id选与被选的次数
{
for (int i = 0; i < tree[id].son_count; i++) {
int temp = tree[id].son.front();
DFS(temp);
tree[id].son.pop_front();
tree[id].chosed *= tree[temp].abandoned;
tree[id].abandoned *= (tree[temp].abandoned + tree[temp].chosed);
}
}
int main() {
int N;
cin >> N;
tree[1].abandoned = 1;
tree[1].chosed = 1;
for (int i = 2; i <= N; i++) {
cin >> tree[i].father;
tree[tree[i].father].son.push_back(i);
tree[tree[i].father].son_count++;
tree[i].chosed = 1;
tree[i].abandoned = 1;
}
DFS(1);
cout << (tree[1].abandoned + tree[1].chosed - 1) % MOD;
return 0;
}