1053 Path of Equal Weight (30分)(树的遍历）

这种题目是重点题型，跟前面刷到的DFS和BFS有很多相同的地方，树的先根遍历对应DFS，树的层序遍历对应BFS，这种题目也有很强的规律可循，刷的多了，就会越来越得心应手了。

题目描述如下：


题目大致意思：
给出一个带权重的树，求出从根节点到叶子节点的路径上的权重和等于给定值的路径。
大致思路：
这道题可以用DFS来做，具体的实现过程如下面的代码所示：
提交结果：
有一个测试用例未能通过，暂时放在第二轮刷题时解决。

提交的代码如下：

#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
struct Node
{
    int weight;
    vector<int> child;
};
Node node[100];
vector<int> temp;
vector<vector<int> >ans;
int n,m,s;
void DFS(int index,int sum);
bool cmp(vector<int> arr1,vector<int> arr2)
{
    for(int i=0;i<arr1.size();i++)
    {
        if(arr1[i]>arr2[i])
            return true;
        if(arr1[i]<arr2[i])
            return false;
    }
}
int main()
{
    cin>>n>>m>>s;
    for(int i=0;i<n;i++)
    {
        cin>>node[i].weight;
    }
    for(int i=0;i<m;i++)
    {
        int id,k;
        cin>>id>>k;
        for(int j=0;j<k;j++)
        {
            int temp;
            cin>>temp;
            node[id].child.push_back(temp);
        }
    }
    temp.push_back(node[0].weight);
    DFS(0,node[0].weight);
    sort(ans.begin(),ans.end(),cmp);
    for(int i=0;i<ans.size();i++)
    {
        for(int j=0;j<ans[i].size()-1;j++)
        {
            cout<<ans[i][j]<<" ";
        }
        cout<<ans[i][ans[i].size()-1]<<endl;
    }
}
void DFS(int index,int sum)
{
    if(node[index].child.size()==0)
    {
        if(sum==s)
            ans.push_back(temp);
        return;
    }
    for(int i=0;i<node[index].child.size();i++)
    {
        temp.push_back(node[node[index].child[i]].weight);
        DFS(node[index].child[i],sum+node[node[index].child[i]].weight);
        temp.pop_back();
    }
}

本次提交后累计得分为1444。