题目详情 - L2-020 功夫传人 (25 分) (pintia.cn)
题解:我的思路是用并查集找到祖师爷,然后进行图的层序遍历。当遇到得道者就把他的功力加起来。(这题中,图的层序遍历要把一层遍历完之后再进行下一层的遍历,这样可以方便计算功力值。)
还是太菜了,要七十多行代码才能解出来
Code Segment:
#include<cstdio>
#include<set>
#include<queue>
#include<cmath>
using namespace std;
const int maxn=1e5+1;
int f[maxn];
queue<int> q,e;
set<pair<int,double>> s[maxn];
int find(int x)
{
while(f[x]!=x)
x=f[x];
return x;
}
int main()
{
int n,j,i,k,t,l,num=0,d;
double z,r,p,sum=0;
scanf("%d%lf%lf",&n,&z,&r);
r/=100;
for(i=0;i<n;i++)
{
scanf("%d",&k);
if(k==0)
{
scanf("%lf",&p);
s[f[i]].erase(make_pair(i,r));
s[f[i]].insert(make_pair(i,p));
num++;
}
else
for(l=0;l<k;l++)
{
scanf("%d",&j);
f[j]=i;
s[i].insert(make_pair(j,r));
}
}
t=find(0);
q.push(t);
if(n==1)
sum+=z*p;
else
while(num)
{
z*=(1-r);
while(!q.empty())
{
d=q.front();
q.pop();
for(auto it=s[d].begin();it!=s[d].end();it++)
{
e.push(it->first);
if(it->second!=r)
{
sum+=z*it->second;
num--;
}
}
}
while(!e.empty())
{
q.push(e.front());
e.pop();
}
}
d=floor(sum);
printf("%d\n",d);
}