第五届蓝桥杯JavaB组第7题

题目

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

考虑所有情况，1.上级加入，本人便不可去，

    2.上级不加入，本人不去，

    3.上级不加入，本人去

分别对每个人进行三种情况的考查。进行递归遍历。

代码如下：

static ArrayList<Integer> a = new ArrayList<Integer>();
static int sum = 0;
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
int[] s = new int
;
for (int i = 1; i < n; i++) {
s[i] = sc.nextInt();
}
s(0,n,s);
System.out.println(sum);
}
public static void s(int code,int n,int[] s){
if(code==n){//到达终点
if(a.size()!=0){//栈不能为空，也就是不能都不去
sum++;
}
return;
}
if(a.contains((Integer)s[code])){
s(code+1,n,s);
}else{
s(code+1,n,s);//不去
a.add(code+1);//加入栈，代表去
s(code+1,n,s);//去
a.remove(a.size()-1);//每次一种情况过后，从后往前出栈，防止影响其他情况
}
}