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;
}