PAT.A1053 Path of Equal Weight

返回目录

题意

给定一棵树和每个结点的权值，求所有从根结点到叶子结点的路径，使得每条路径上结点的权值之和等于给定的常数S。如果有多条这样的路径，按路径非递增的顺序输出。其中路径的大小是指，如果两条路径分别为a1→a2→…→an与b1→b2→…→bm,且有a1== b1、 a2== b2、 … ai-1==bi-1成立， 但ai>bi, 那么称第一条路径比第二条路径大。

样例（可复制）

20 9 24
10 2 4 3 5 10 2 18 9 7 2 2 1 3 12 1 8 6 2 2
00 4 01 02 03 04
02 1 05
04 2 06 07
03 3 11 12 13
06 1 09
07 2 08 10
16 1 15
13 3 14 16 17
17 2 18 19

//output
10 5 2 7
10 4 10
10 3 3 6 2
10 3 3 6 2

注意点

1. 注意只有到达叶节点时sum==s才输出
2. 保存路径可以使用一个数组或vector，如果使用了vector记得最后节点返回时要pop_back()在path的最后那个节点

#include<bits/stdc++.h>
using namespace std;

int n,m,s;//节点总数，非叶节点数，给定的和 
int path[110];
struct Node{
	int data;
	vector<int> child;
}node[110]; 
bool cmp(int a,int b){
	return node[a].data>node[b].data;
}
void DFS(int v,int num,int sum){
	if(sum>s)return;
	if(sum==s){
		if(node[v].child.size()!=0)return;
		for(int i=0;i<num;i++){
			printf("%d",node[path[i]].data);
			if(i!=num-1)printf(" ");
			else printf("\n");
		}
		return;
	}
	for(int i=0;i<node[v].child.size();i++){
		int child=node[v].child[i];
		path[num]=child;
		DFS(child,num+1,sum+node[child].data); 
	}
}
int main(){
	cin>>n>>m>>s;
	for(int i=0;i<n;i++)scanf("%d",&node[i].data);
	int id,num,tmp;//编号，数量，孩子节点 
	while(m--) {
		scanf("%d%d",&id,&num); 
		while(num--){
			scanf("%d",&tmp);
			node[id].child.push_back(tmp);
		}
		sort(node[id].child.begin(),node[id].child.end(),cmp);
	}
	path[0]=0;
	DFS(0,1,node[0].data);//根节点，节点数量，总和 
    return 0;
}