1053 Path of Equal Weight (30 分)

最新推荐文章于 2022-04-13 00:26:50 发布

Jcldcdmf

最新推荐文章于 2022-04-13 00:26:50 发布

阅读量795

点赞数 3

CC 4.0 BY-SA版权

分类专栏： PTA

本文链接：https://blog.youkuaiyun.com/qq_37613112/article/details/101107444

PTA 专栏收录该内容

54 篇文章

订阅专栏

本文深入探讨了深度优先搜索(DFS)算法的原理与应用，通过一个具体的树形结构问题，详细介绍了如何利用DFS进行有效的路径查找。文章提供了完整的C++代码实现，包括节点结构定义、递归遍历过程及优化技巧，如子节点排序以提高遍历效率。适合初学者理解DFS的基本概念和实际操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

DFS

类似的题目还有A1103

胡凡和曾磊的算法笔记对于类似的题目讲的很明白，做完第八章的几道题应该就掌握了。

标题沿着一条路往下走，当已经能够判断不满足条件时返回，当满足条件时，完成相应操作后返回，在遍历的过程中，对于一个结点有选或不选两种选择，相应的就会有两条路，在本题中，由于树形结构不经过父节点无法到达子节点，所以当不选某个节点时就没有必要再往下走了，因此只需要写选择这个节点时的代码。

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#define maxn 100
int n,m;
long long int s;
struct node{
    int weight;
    vector<int>next;
}node[maxn];
vector<int>temp;
bool cmp(int a,int b){//让权值大的子节点排到前边，以便于遍历时先被选出
    return node[a].weight>node[b].weight;
}
void dfs(int index,int sum){//也可以用数组代替temp，这时需要为数组设置一个下标
    if(sum==s){
        if(node[index].next.size()!=0)return;
        for (int i = 0; i < temp.size(); ++i) {
            if(i<temp.size()-1)cout<<temp[i]<<" ";
            if(i==temp.size()-1)cout<<temp[i]<<endl;
        }
    }
    if(index>=n||sum>s)return;
    for (int i = 0; i < node[index].next.size(); ++i) {
        temp.push_back(node[node[index].next[i]].weight);
        dfs(node[index].next[i],sum+node[node[index].next[i]].weight);
        temp.pop_back();
    }
}
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;
        cin>>id;
        int size;
        cin>>size;
        for (int j = 0; j < size; ++j){
            int child;
            cin>>child;
            node[id].next.push_back(child);
        }
        sort(node[id].next.begin(),node[id].next.end(),cmp);
    }
    temp.push_back(node[0].weight);
    dfs(0,node[0].weight);
    return 0;
}