PAT1053 树DFS

题目:

根据节点的值,从大到小输出值为S的所有路径。

思路:这题主要是如何保存路径麻烦。节点从大到小在输入的时候就能够处理。然后每个节点用一个vector保存路径。

#include<iostream>
#include<vector>
#include<algorithm>
#include<stdio.h>
using namespace std;
struct node
{
    int all,value;
    vector<int>child;//存放子节点的下标
    vector<int>path;
};
node lists[200];
bool cmp(int x,int y)
{
    return lists[x].value>lists[y].value;
}
void dfs(int index,int S)
{
    if(lists[index].all>S)
        return;
    if(lists[index].all==S)
    {
        if(lists[index].child.size()!=0)
            return;
        int num=lists[index].path.size();
        for(int i=0;i<num;i++)
        {
            int j=lists[index].path[i];
            if(i==num-1)
                printf("%d\n",lists[j].value);
            else
                printf("%d ",lists[j].value);
        }
    }
    //当前all<S
    int numchild=lists[index].child.size();
    for(int i=0;i<numchild;i++)
    {
        int x=lists[index].child[i];
        lists[x].all=lists[index].all+lists[x].value;
        lists[x].path.clear();
        lists[x].path=lists[index].path;
        lists[x].path.push_back(x);
        dfs(x,S);
    }
    return;
}
int main()
{
    int N,M,S;

    cin>>N>>M>>S;
    for(int i=0;i<N;i++)
        {
        cin>>lists[i].value;
        }
    for(int i=0;i<M;i++)
    {
        int x,num,y;
        cin>>x>>num;
        for(int j=0;j<num;j++)
        {
            cin>>y;
            lists[x].child.push_back(y);
        }
        sort(lists[x].child.begin(),lists[x].child.end(),cmp);
    }
    //初始化根节点
      lists[0].all=lists[0].value;
      lists[0].path.push_back(0);
      dfs(0,S);
    return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值