题记
思路:
1.这道题是普通的树,建立结点的结构体,用vector来存储孩子结点的下标
2.用vector来存储当前得到的路径,递归的前依次把当前结点的孩子加进容器,递归后记得pop_back()。
3.进入dfs,参数为index,sum分别指当前结点的下标和当前路径的权值和,如果sum>s直接return;如果sum==s首先判断当前结点是不是叶子,若不是return,若是依次输出路径上各结点的权重;如果sum<s那么进行循环,分别把当前结点的孩子加进path中进行递归,记得递归完path.pop_back()!!!
4.本题要求输出按照非递增,那么有一个很巧妙的方法,输入一个非叶子结点的child之后,直接进行排序,总是让权重较大的结点放在前面,那么最后直接按照递归的先后顺序输出path就能得到满足题目要求的顺序。因为每一步我们都是从大到小加到path里的。
代码如下
#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
const int Maxn=100+5;
struct node{
int weight;
//存放孩子的编号
vector<int>child;
}Node[Maxn];
bool cmp(int a,int b){
return Node[a].weight>Node[b].weight;
}
//记录路径
vector<int>path;
//输入
int n,m,s;
//当前访问到下标为index的结点,目前的和为sum
void dfs(int index,int sum){
if(sum>s)
return;
if(sum==s){//如果现在的路径的和已经是s了
//虽然权值的和满足条件,但是当前结点不是叶子
if(Node[index].child.size()!=0)
return;
else{
for(int i=0;i<path.size();i++){
printf("%d",path[i]);
if(i<path.size()-1)
printf(" ");
else
printf("\n");
}
}
return;
}
//路径上的权值和小于s
for(int i=0;i<Node[index].child.size();i++){
path.push_back(Node[Node[index].child[i]].weight);
dfs(Node[index].child[i],sum+Node[Node[index].child[i]].weight);
path.pop_back();
}
}
int main()
{
scanf("%d %d %d",&n,&m,&s);
for(int i=0;i<n;i++)
scanf("%d",&Node[i].weight);
for(int i=0;i<m;i++){
int index,k,child;
scanf("%d %d",&index,&k);
for(int j=0;j<k;j++){
scanf("%d",&child);
Node[index].child.push_back(child);
}
//该结点的孩子排序,之后保存的路径按先后顺序输出就可以保证从大到小
sort(Node[index].child.begin(),Node[index].child.end(),cmp);
}
path.push_back(Node[0].weight);
dfs(0,Node[0].weight);
return 0;
}