PTA——功夫传人(图论-图的遍历)

• 基本思路
  用一个数组记录得道者功力值的倍数，再用一个bool数组标记该节点是否是得道者，还需要一个double数组记录功力值。最后bfs或者dfs来求得每个节点的功力值

#include<bits/stdc++.h>
using namespace std;
const int N = 100010;
bool is[N];//是不是得道者
vector<int> edge[N]; 
int n,multiple[N],q[N];//倍数 
double w[N];//存每个人的功力值 
bool vis[N];
double z,r;
double ans;
inline void dfs(int x){
	vis[x]=true;
	for(auto y:edge[x]){
		if(!vis[y]){
			w[y]=w[x]-(w[x]*r*0.01);
			dfs(y);
		}
	}
}
inline void bfs(int s){
	int front=1,rear=0;//用数组模拟队列 
	q[++rear]=s;//起始点入队 
	vis[s]=true;//标记 
	while(front<=rear){//队列不为空 
		int x=q[front];//取出队头 
		++front;//出队 
		for(auto y:edge[x]){//遍历x的每个子节点y 
			if(!vis[y]){
				vis[y]=true;
				w[y]=w[x]-(w[x]*r*0.01);//y的功力值 是他师傅功力值减弱r%后的值 
				q[++rear]=y;//将子节点入队 
			}
		}
	}
}

int main(){
	cin>>n>>z>>r;
	for(int i=0;i<n;i++){
		int k; cin>>k;
		if(k==0){
			is[i]=true;
			int id; cin>>id;
			multiple[i]=id;
		}else{
			for(int j=1;j<=k;j++){
				int x;  cin>>x;
				edge[i].push_back(x);
			}
		}
	}
	w[0]=z;
	//dfs(0);
	bfs(0);//祖师爷的编号为0

	for(int i=0;i<n;i++){
		if(is[i])//是得道者 
		  ans+=w[i]*multiple[i]; 
	} 
	cout<<(int)ans;
	return 0; 
}