L2-020. 功夫传人(dfs+vector 或者 邻接矩阵+dij+优先队列）

L2-020. 功夫传人

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

一门武功能否传承久远并被发扬光大，是要看缘分的。一般来说，师傅传授给徒弟的武功总要打个折扣，于是越往后传，弟子们的功夫就越弱…… 直到某一支的某一代突然出现一个天分特别高的弟子（或者是吃到了灵丹、挖到了特别的秘笈），会将功夫的威力一下子放大N倍 —— 我们称这种弟子为“得道者”。

这里我们来考察某一位祖师爷门下的徒子徒孙家谱：假设家谱中的每个人只有1位师傅（除了祖师爷没有师傅）；每位师傅可以带很多徒弟；并且假设辈分严格有序，即祖师爷这门武功的每个第i代传人只能在第i-1代传人中拜1个师傅。我们假设已知祖师爷的功力值为Z，每向下传承一代，就会减弱r%，除非某一代弟子得道。现给出师门谱系关系，要求你算出所有得道者的功力总值。

输入格式：

输入在第一行给出3个正整数，分别是：N（<=10⁵）——整个师门的总人数（于是每个人从0到N-1编号，祖师爷的编号为0）；Z——祖师爷的功力值（不一定是整数，但起码是正数）；r ——每传一代功夫所打的折扣百分比值（不超过100的正数）。接下来有N行，第i行（i=0, ..., N-1）描述编号为i的人所传的徒弟，格式为：

K_i ID[1] ID[2] ... ID[K_i]

其中K_i是徒弟的个数，后面跟的是各位徒弟的编号，数字间以空格间隔。K_i为零表示这是一位得道者，这时后面跟的一个数字表示其武功被放大的倍数。

输出格式：

在一行中输出所有得道者的功力总值，只保留其整数部分。题目保证输入和正确的输出都不超过10¹⁰。

输入样例：

10 18.0 1.00
3 2 3 5
1 9
1 4
1 7
0 7
2 6 1
1 8
0 9
0 4
0 3

输出样例：

404
邻接矩阵+dij+优先队列

#include<iostream>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<cmath>
#define inf 0x3f3f3f3f
using namespace std;

struct node
{
    int v;
    int nxt;
}e[100001];
int head[100001];
int f[100001];
int cnt=1;
int a[100001];//用来放得道者
int b[100001];//用来放得道者的放大倍数
int d[100001];//到祖师的最短距离
double sum=0;
    int n;
void add(int u,int v)//邻接表
{
    e[cnt].v=v;
    e[cnt].nxt=head[u];
    head[u]=cnt++;
}
struct cmp
{
    bool operator()(int x,int y)
    {
        return x>y;
    }
};

priority_queue<int,vector<int>,cmp>q;
void dijstra(int s)
{
    memset(f,0,sizeof f);
    d[s]=0;
    f[s]=1;
    for(int i=head[s];i!=-1;i=e[i].nxt )
    {
        int v=e[i].v;
        d[v]=1;
        f[v]=1;
        q.push(v);
    }
    while(!q.empty())//挑出最小的松弛
    {
        int k=q.top();q.pop();
        for(int i=head[k];i!=-1;i=e[i].nxt)
        {
            int v=e[i].v;
            if(d[v]>d[k]+1)
            {
                d[v]=d[k]+1;
                q.push(v);
            }
        }
    }
}

int main()
{

    int p=1;
    double z,r;
    cin>>n>>z>>r;
    memset(head,-1,sizeof head);
    memset(d,inf,sizeof d);
    for(int i=0;i<n;i++)
    {
        int k;
        cin>>k;
        if(k==0)
        {
            int x;
            cin>>x;
            a[p]=i;
            b[p++]=x;
        }
        else
        {
            for(int j=1;j<=k;j++)
            {
                int x;
                cin>>x;
                f[x]=1;
                add(i,x);
            }
        }
    }
    int zs;
    for(int i=0;i<n;i++)
    {
        if(!f[i])
        {
            zs=i;
            break;
        }
    }
    dijstra(zs);
    r=(100-r)/100;
    for(int i=1;i<=p-1;i++)//计算
    {
        int x=a[i];
        double ss;
        ss=z*pow(r,d[x]);
        sum+=ss*b[i];
    }
    cout<<(int)sum;

}

dfs+vector

#include<iostream>
#include<cstring>
#include<string>
#include<queue>
#include<algorithm>
#include<cmath>
#include<vector>
#define inf 0x3f3f3f3f
using namespace std;
int n;
double z,r;
vector<int>v[100005];
int a[100005];

double dfs(int x,int t)  
{  
    double s=0;  
    if(v[x].size()==0) s+=a[x]*z*pow(r,t);  
    else   
        for(int i=0;i<v[x].size();i++)  
            s+=dfs(v[x][i],t+1);  
    return s;  
} 


int main()
{
    cin>>n>>z>>r;
    r=(100-r)/100;
    for(int i=0;i<n;i++)
    {
        int k;
        cin>>k;
        if(k!=0)
        {
            for(int j=1;j<=k;j++)
            {
                int x;
                cin>>x;
                v[i].push_back (x);
            }
        }
        else
        {
            cin>>a[i];
        }
    }
    int s=dfs(0,0);
    cout<<s;
}