PTA 功夫传人(bfs+dfs)

BFS

#include<vector>
#include<bits/stdc++.h>
using namespace std;
int a[100005];
map<int,int>vis;
int n,m,x;
double z,r;
vector<int>v[100006];
struct p{
	int data;
	double sum;
}mu;
double sum=0;
int bfs(int ind,double zhi){
	queue<p>q;									//用队列存每个人的编号和功力		
	q.push({ind,zhi});							//先压入初始值
	while(!q.empty()){
		mu=q.front();							//将队首的人取出,开始计算
		q.pop();
		if(vis[mu.data]){
			sum+=(mu.sum)*vis[mu.data];			//如果他是得道者,直接求出功力
			continue;							//因为得道者没有徒弟,直接开始下一次循环
		}
		for(int i=0;i<v[mu.data].size();i++){
			q.push({v[mu.data][i],mu.sum*r});	//将每个人的编号和功力值压入
		}
	}
}
int main()
{
	cin>>n>>z>>r;
	r=(1-r/100);				//方便传功时求值
	for(int i=0;i<n;i++){
		cin>>m;
		if(m==0){
			cin>>x;
			vis[i]=x; 			//把每个得道者标记
		}
		for(int j=0;j<m;j++){
			cin>>x;
			v[i].push_back(x);//用vector存每个人的徒弟的编号
		}
	}
	bfs(0,z);
	
	printf("%.d\n",int(sum));	//输出sum
	return 0;
}

DFS

#include<bits/stdc++.h>
using namespace std;
int n,m,x;
double vis[100005];
vector<int>v[100005];
double r,z,sum=0;
void dfs(int ind,double zhi){
	if(vis[ind]){
		sum+=vis[ind]*zhi;				//当编号为ind的人是得道者的时候,求他们功力的和
	}
	for(int i=0;i<v[ind].size();i++){
		dfs(v[ind][i],zhi*r);			//跑每个人的徒弟,分别求值,其他都一样
	}
	return;
}
int main()
{
	cin>>n>>z>>r;
	r=(1-r/100);
	for(int i=0;i<n;i++){
		cin>>m;
		if(m==0){
			cin>>x;
			vis[i]=x;
		}
		for(int j=0;j<m;j++){
			cin>>x;
			v[i].push_back(x);
		}
	}
	dfs(0,z);
	cout<<int(sum)<<endl;
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值