这道题大一上学期校内天梯选拔的时候遇到过,但是没有学过数据结构,就是再用循环和一个非常大的二位数组去做,结果有几个测试点段错误,还有运行超时之类的,得了十几分后来这道题就没怎么看过,现在该上大二了,再看这道题,感觉是真的简单,直接BFS广搜一遍就行了。
由于题目保证了每个徒弟只有一个师傅,这就致使从祖宗往下进行广度优先搜索时,BFS的深度就是徒弟的代数,定义一个level[]数组表示第几层,祖宗的徒弟标记为1,每次出队时,让该元素的徒弟的代数=该元素的代数+1,即level[徒弟]=level[师傅]+1;如果BFS过程中遇到了得道者,直接计算它的功力值,即祖宗的功力值代数(1-折扣%)得道倍数;将它加入到sum中这样这道题就能方便的做出来了
需要注意的一点是,第二个测试点(即测试点1)是祖宗就是得道者的情形,这种情况直接分开分析即可,即if(祖宗是得道者){sum=祖宗功力值得道倍数;return 0;}
代码如下:
#include<iostream>
#include<vector>
#include<queue>
#include<cmath>
using namespace std;
vector<int> v[100005];
int level[100005];
int getdao[100005];
int main() {
int n;
double z,r;
cin>>n>>z>>r;
r=1-r/100;
for(int i=0; i<n; i++) {
int k,t;
scanf("%d",&k);
if(k==0)
{
int bei;
scanf("%d",&bei);
getdao[i]=bei;
}
for(int j=0; j<k; j++) {
scanf("%d",&t);
v[i].push_back(t);
}
}
double sum=0;
queue<int> q;
if(getdao[0]!=0)
{
printf("%d",(int)(z*getdao[0]));
return 0;
}
for(int i=0;i<v[0].size();i++)
{
q.push(v[0][i]);
level[v[0][i]]=1;
}
while(!q.empty())
{
int t=q.front();q.pop();
// cout<<t<<' ';
if(getdao[t]!=0)
{
sum+=z*pow(r,level[t])*getdao[t];
// cout<<getdao[t]<<' ';
// cout<<z*1.0*pow(r,level[t])*getdao[t]<<' ';
// cout<<level[t];;
continue;
}
for(int i=0;i<v[t].size();i++)
{
int p=v[t][i];
level[p]=level[t]+1;
q.push(p);
// cout<<p;
}
}
printf("%d",(int)sum);
}