PATA 1053 Path of Equal Weight

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
题记
思路:
1.这道题是普通的树,建立结点的结构体,用vector来存储孩子结点的下标
2.用vector来存储当前得到的路径,递归的前依次把当前结点的孩子加进容器,递归后记得pop_back()。
3.进入dfs,参数为index,sum分别指当前结点的下标和当前路径的权值和,如果sum>s直接return;如果sum==s首先判断当前结点是不是叶子,若不是return,若是依次输出路径上各结点的权重;如果sum<s那么进行循环,分别把当前结点的孩子加进path中进行递归,记得递归完path.pop_back()!!!
4.本题要求输出按照非递增,那么有一个很巧妙的方法,输入一个非叶子结点的child之后,直接进行排序,总是让权重较大的结点放在前面,那么最后直接按照递归的先后顺序输出path就能得到满足题目要求的顺序。因为每一步我们都是从大到小加到path里的。

代码如下

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

const int Maxn=100+5;

struct node{
    int weight;
    //存放孩子的编号
    vector<int>child;
}Node[Maxn];

bool cmp(int a,int b){
    return Node[a].weight>Node[b].weight;
}

//记录路径
vector<int>path;

//输入
int n,m,s;

//当前访问到下标为index的结点,目前的和为sum
void dfs(int index,int sum){
    if(sum>s)
        return;
    if(sum==s){//如果现在的路径的和已经是s了
        //虽然权值的和满足条件,但是当前结点不是叶子
        if(Node[index].child.size()!=0)
            return;
        else{
            for(int i=0;i<path.size();i++){
                printf("%d",path[i]);
                if(i<path.size()-1)
                    printf(" ");
                else
                    printf("\n");
            }
        }
        return;
    }
    //路径上的权值和小于s
    for(int i=0;i<Node[index].child.size();i++){
        path.push_back(Node[Node[index].child[i]].weight);
        dfs(Node[index].child[i],sum+Node[Node[index].child[i]].weight);
        path.pop_back();
    }
}

int main()
{
    scanf("%d %d %d",&n,&m,&s);
    for(int i=0;i<n;i++)
        scanf("%d",&Node[i].weight);
    for(int i=0;i<m;i++){
        int index,k,child;
        scanf("%d %d",&index,&k);
        for(int j=0;j<k;j++){
            scanf("%d",&child);
            Node[index].child.push_back(child);
        }
        //该结点的孩子排序,之后保存的路径按先后顺序输出就可以保证从大到小
        sort(Node[index].child.begin(),Node[index].child.end(),cmp);
    }
    path.push_back(Node[0].weight);
    dfs(0,Node[0].weight);

    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值