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
数据规模与约定
对于20%的数据,n ≤ 20;
对于40%的数据,n ≤ 100;
对于100%的数据,1 ≤ n ≤ 100000。
资源约定:
峰值内存消耗 < 256M
CPU消耗 < 2000ms
请严格按要求输出,不要画蛇添足地打印类似:“请您输入...” 的多余内容。
所有代码放在同一个源文件中,调试通过后,拷贝提交该源码。
#include<stdio.h>
#include<string.h>
int a[100001],b[100001];
int n,t;
int str(int i)
{
if(i>n) //假如已经将n名士兵遍历一遍,就加一
{
t++;
return t;
}
if(a[b[i]]==0) //假如第i名士兵的直接上级已经出列
{
str(i+1); //那么这名士兵就不能出列,直接跳到下一位
}
else //假如第i名士兵的直接上级没有出列
{
a[i]=0; //让他出列并跳到下一位
str(i+1);
a[i]=i; //不让他出列并跳到下一位
str(i+1);
}
}
int main()
{
int i,j;
while(scanf("%d",&n)!=EOF)
{
t=0;
memset(a,0,sizeof(a));
memset(b,0,sizeof(b));
a[0]=1;
for(i=1;i<=n;i++)
{
a[i]=i; //储存当前状态,如果已经出列就置为0
}
b[1]=0;
for(i=2;i<=n;i++)
{
scanf("%d",&b[i]); //记录第i号士兵的直接上级
}
printf("%d\n",(str(1)-1)%10007); //因为要排除没有人去的情况,所以要减一
}
return 0;
}
刚开始总是出错,实在没办法了就上网搜资料,后来发现方法思路是没有错的,只不过是一些细节上没有处理好,主要是对递归的运用不够熟练,后来理清思路之后又改正了。我也看到有其他的大神是用dp写的,但是本人对于dp一直以来都没有弄明白,所以就只能用递归来写。各位将就着看吧^_^ ^_^