PAT A1053 Path of Equal Weight (30 分) 树遍历 DFS

最新推荐文章于 2022-07-12 18:30:56 发布

原创最新推荐文章于 2022-07-12 18:30:56 发布 · 89 阅读

0 ·

CC 4.0 BY-SA版权

PAT 专栏收录该内容

165 篇文章

订阅专栏

本文介绍了一种基于深度优先搜索（DFS）的算法，用于在一棵树形结构中寻找所有从根节点到叶子节点的数据域和等于特定值S的路径，并使用C++实现。该算法将所有满足条件的路径存储在vector容器中，最后使用greater比较器进行降序排序并输出。

题目大意：给出一棵树，找到所有满足数据域和为S的从根节点到叶子节点的路径，并按降序输出所有的序列。

dfs方法搜索所有的路径，存储到vector<vector<int>>中，然后用greater<vector<int>>()进行排序输出即可。

AC代码：

#include <vector>
#include <algorithm>
#include <cstdio>
#include <set>

using namespace std;

struct Node
{
    int data;
    vector<int> child;
};

void dfs(int index, int sum, int S, vector<Node> &tree, vector<int> &tmp, vector<vector<int>> &ans)
{
    if(tree[index].child.size() == 0)
    {
        if(sum == S) ans.push_back(tmp);
        return;
    }
    for (int i = 0; i < tree[index].child.size(); ++i)
    {
        int child = tree[index].child[i];
        tmp.push_back(tree[child].data);
        dfs(child, sum + tree[child].data, S, tree, tmp, ans);
        tmp.pop_back();
    }
}

int main()
{
    int N, M, S;
    scanf("%d%d%d", &N, &M, &S);
    vector<Node> tree(N);
    for (int i = 0; i < N; ++i)
        scanf("%d", &tree[i].data);
    for (int i = 0; i < M; ++i)
    {
        int index, childNum;
        scanf("%d%d", &index, &childNum);
        for (int j = 0; j < childNum; ++j)
        {
            int child;
            scanf("%d", &child);
            tree[index].child.push_back(child);
        }
    }
    vector<vector<int>> ans;
    vector<int> tmp;
    tmp.push_back(tree[0].data);
    int sum = tree[0].data;
    dfs(0,sum, S, tree, tmp, ans);
    sort(ans.begin(), ans.end(), greater<vector<int>>());
    for (int i = 0; i < ans.size(); ++i)
    {
        for (int j = 0; j < ans[i].size(); ++j)
        {
            printf("%d", ans[i][j]);
            if(j < ans[i].size() - 1) printf(" ");
            else printf("\n");
        }
    }
    return 0;
}