1053. Path of Equal Weight (30)

题目链接：https://www.patest.cn/contests/pat-a-practise/1053

题目大意：给定一个权值和sum，在所有从树的根节点到叶子结点的路径中，选出路径的权值和等于sum的所有路径。

思路：深度优先遍历所有路径，找出满足条件的路径。注意DFS函数的结束条件

代码如下(有详细注释)

#include <iostream>#include <cstdio>#include <algorithm>#include <vector>using namespace std;struct node{    int w;//权值    vector<int> child;//孩子节点的下标};int sum;//路径和vector<node> v;//记录树的每个节点，下标为序号vector<int> path;//元素值为节点的序号，也即在v中的下标void DFS(int index,int pathcount,int currentsum){    /*深度优先地遍历整棵树。    index为当前数的根节点的下标;    pathcount为当前节点属于当前路径的第几个节点    sum为当前路径的部分权值和*/    if(currentsum>sum)//当前权值和超出要求        return;    if(currentsum==sum){//权值和已满足要求        if(v[index].child.size()!=0)//但该节点不是叶子结点            return;        for(int i=0;i<pathcount;i++){            if(i==pathcount-1)                cout<<v[path[i]].w<<endl;            else                cout<<v[path[i]].w<<" ";        }    }    /*权值和还未达到目标值，深度优先地遍历该节点的每一个孩子*/    for(int i=0;i<v[index].child.size();i++){        int tmp=v[index].child[i];        path[pathcount]=tmp;        DFS(tmp,pathcount+1,currentsum+v[tmp].w);    }}bool cmp(int a,int b){    return v[a].w>v[b].w;}int main(int argc, char const *argv[]){    int n,m;    cin>>n>>m>>sum;    v.resize(n);    path.resize(n);    for(int i=0;i<n;i++)        cin>>v[i].w;    for(int i=0;i<m;i++){        int tmp,cnt;        scanf("%02d %d",&tmp,&cnt);        v[tmp].child.resize(cnt);        for(int j=0;j<cnt;j++){            scanf("%02d",&v[tmp].child[j]);        }        sort(v[tmp].child.begin(),v[tmp].child.end(),cmp);//此处排序是为了满足输出要求    }    DFS(0,1,v[0].w);    return 0;}