L2-020 功夫传人 (25分) C++ dfs解法

最新推荐文章于 2024-05-31 11:51:21 发布

Jay_fearless

最新推荐文章于 2024-05-31 11:51:21 发布

阅读量340

点赞数

CC 4.0 BY-SA版权

分类专栏： PTA刷题文章标签： dfs 算法数据结构

本文链接：https://blog.youkuaiyun.com/Jay_fearless/article/details/109361125

PTA刷题专栏收录该内容

90 篇文章

订阅专栏

本文介绍了一种算法，用于计算一个武学门派中所有得道者的武功总值。通过建立师徒关系的数据结构，并考虑每代传承时武功值的衰减，最终输出所有得道者的武功值总和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

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

输入说明

输入在第一行给出3个正整数，分别是：N（≤10
5
）——整个师门的总人数（于是每个人从0到N−1编号，祖师爷的编号为0）；Z——祖师爷的功力值（不一定是整数，但起码是正数）；r ——每传一代功夫所打的折扣百分比值（不超过100的正数）。接下来有N行，第i行（i=0,⋯,N−1）描述编号为i的人所传的徒弟，格式为：
Ki ID[1] ID[2] ⋯ ID[Ki]
其中Ki 是徒弟的个数，后面跟的是各位徒弟的编号，数字间以空格间隔。Ki 为零表示这是一位得道者，这时后面跟的一个数字表示其武功被放大的倍数。

输出说明

在一行中输出所有得道者的功力总值，只保留其整数部分。题目保证输入和正确的输出都不超过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

输出样例

22分代码

之前使用while循环寻找父结点和其到子节点之间的距离，虽然不需要存图，但是对于时间的消耗太大，最后一个测试点显示运行超时，结果只有22分……

#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> PII;
vector<PII> V;
const int N = 1e5+10;
int n,fa[N];
double Z,r,ans; 
int main()
{
    scanf("%d%lf%lf",&n,&Z,&r);
    for(int i=0;i<n;i++) fa[i]=i;
	int k,t;
    for(int i=0;i<n;i++)
    {
    	scanf("%d",&k);
    	if(k!=0)
    	{
    		for(int j=0;j<k;j++)
    		{
				scanf("%d",&t);	    			
				fa[t]=i;
			}
		}
		else{
			scanf("%d",&t);
			V.push_back({i,t});
		}
	}
	for(int i=0;i<V.size();i++)
	{
		auto t=V[i].second,ii=V[i].first;
		int dis=0,temp=ii;
		while(temp!=0)
		{
			dis++;
			temp=fa[temp];
		}
		ans+=Z*pow(1-r/100,dis)*t;
	}
	cout<<int(ans);
    return 0;
}

后来使用vector数组存储师傅和徒弟关系进行dfs就可以AC了。

AC代码

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
vector<int> V[N];
int vis[N],n;
double g[N];
double z,r,ans;
void dfs(int u)
{
	if(V[u].size()<1) return ;
    for(int i=0;i<V[u].size();i++)
    {
        double ss=g[u]*(100-r)/100;
        if(vis[V[u][i]])
        {
            ans+=ss*vis[V[u][i]];   
        }
        else
        {
            g[V[u][i]]=ss;
            dfs(V[u][i]);
        }
    }
}
int main()
{
    cin>>n>>z>>r;
    for(int i=0;i<n;i++)
    {
        int m,a;
        cin>>m;
        if(m==0)
        {
            cin>>a;
            vis[i]=a;
        }
        else
        {
            while(m--)
            {
                cin>>a;
                V[i].push_back(a);
            }
        }
    }
    if(n==1)
    {
        ans=vis[0]*z; 
    }
    else
    {
        g[0]=z;
        dfs(0);
    }
    cout<<(int)ans;
}