问题描述 华为公司积极开拓北美市场,首先在北美建立销售总部,总部在附近地区发展一些销售点,这些销售点可以发展建立下一级销售点,依此类推,最终形成一个新型分级销售网络。假设在销售网络中,有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;
}