1053 Path of Equal Weight (30 分)

最新推荐文章于 2024-10-17 16:41:30 发布

vs5

最新推荐文章于 2024-10-17 16:41:30 发布

阅读量147

点赞数 1

分类专栏： PAT C++ 文章标签：深度优先 c++

本文链接：https://blog.csdn.net/qq_55591643/article/details/123007670

版权

C++ 同时被 2 个专栏收录

50 篇文章 1 订阅

订阅专栏

PAT

44 篇文章 0 订阅

订阅专栏

该程序实现了一个从根节点到叶子节点路径权值和等于给定值w的搜索算法。它使用深度优先搜索策略，并将找到的路径存储在二维向量中，然后按降序排列并输出。程序首先读取树的节点数、边数和目标和，接着读取每个节点的权重和边的信息，最后从根节点开始搜索符合条件的路径。

摘要由CSDN通过智能技术生成

求从根节点到叶子节点路径权值和等于w的全部路径

单调递减输出，我们直接把路径上的权值存到二维vector里，对vector进行排序。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
int n,m,w,r,k,x,v[1010],cnt,idx;
vector<int>e[1010],ans[1010],res;
void dfs(int u,int s)
{   
    if(s > w) return ;
    if(s == w && !e[u].size())
    {   
        ans[idx ++] = res;
        return ;
    }
    for(auto it : e[u])
    {   
        res.push_back(v[it]);
        dfs(it,s + v[it]);
        res.pop_back();
    }
}
int main()
{
    cin >> n >> m >> w;
    for(int i = 0; i < n; i ++) cin >> v[i];
    for(int i = 0; i < m; i ++)
    {
        cin >> r >> k;
        while(k --)
        {
            cin >> x;
            e[r].push_back(x);
        }
    }
    res.push_back(v[0]);
    dfs(0,v[0]);
    sort(ans,ans + idx,[](vector<int> a,vector<int> b)
    {
        return a > b;
    });
    for(int i = 0; i < idx; i ++)
    {
        for(int j = 0; j < ans[i].size(); j ++)
        {
            if(j != 0) cout << ' ';
            cout << ans[i][j];
        }
        cout << '\n';
    }
    return 0;
}