一道华为机试题-销售点撤销方案数

问题描述 华为公司积极开拓北美市场，首先在北美建立销售总部，总部在附近地区发展一些销售点，这些销售点可以发展建立下一级销售点，依此类推，最终形成一个新型分级销售网络。假设在销售网络中，有N个销售点（包括总部），将它们分别编号为1至N。考虑到金融危机，销售总部决定撤销一些销售点，保留其他销售点。需要注意是： （1）如果撤销一个销售点，那么该销售点发展的所有下级销售点均要撤销，依此类推； （2）销售总部不能撤销自己； （3）销售总部可以不撤销任何销售点。 请你帮忙告诉华为公司：共存在多少个销售点撤销方案。

问题输入： 输入包括多个行，首先给出一个整数N，接着N-1行给出销售网络的建立过程，在这N-1行中，第j行（1≤j≤N-1）给出一个整数k（j<k），表示销售点k发展了销售点j。销售点N就是销售总部。

问题输出： 输出一行，给出销售点撤销方案数。

 

 

建一颗树，对于根节点，方案数为子节点方案数的乘积；对于叶子节点，方案数为2；对于非根节点且非叶子节点，方案数为子节点方案数乘积再加1。

#include <iostream>
#include <cmath>
#include <iomanip>
#include <string>
#include <cstring>
#include <vector>
#include <map>
#include <set>
#include <algorithm>
#include <stdio.h>
#include <stack>
#include <queue>
#include<cstdio>
using namespace std;

int n;
vector< vector<int> > vec;

int dfs(int num)
{
	if (num == n)		//根节点
	{
		int ret = 1;
		for (int i=0; i<vec[num].size(); i++)
		{
			ret *= dfs(vec[num][i]);
		}
		return ret;
	}
	if (vec[num].size() == 0)	//叶子节点
	{
		return 2;
	}
	else
	{
		int ret = 1;
		for (int i=0; i<vec[num].size(); i++)
		{
			ret *= dfs(vec[num][i]);
		}
		return ++ret;
	}
}

int main()
{
	while(cin>>n)
	{
		vec.clear();
		for (int i=0; i<=n; i++)
		{
			vector<int> t;
			vec.push_back(t);
		}
		for (int j=1; j<=n-1; j++)
		{
			int k;
			cin>>k;
			vec[k].push_back(j);
		}
		cout<<dfs(n) <<endl;
	}
	return 0;
}