PAT_1042 Path of Equal Weight

题目描述:

给定一棵根节点为R的加权树(每个节点Ti具有权重Wi)。从R到L的路径的权重被定义为沿着从R到任何叶节点L的路径的所有节点的权重之和。.现在给出任何加权树,你应该找到所有的路径他们的权重等于给定的数字。

输入:

第一行:节点总数N,非叶子节点数M,给定的数字S
第二行:从0~N-1节点号的权重
后续M行:ID(节点id),K(节点id的孩子数),ID[1] ID[2] ... ID[K](孩子节点id)

输出:

输出路径值为给定数字S的路径(经过的每个节点的权值),并且路径需要按序输出。
note:如果存在1 <= k <min {n,m}使得Ai = Bi,则序列{A1,A2,...,An}被称为大于序列{B1,B2,...,Bm}。 i = 1,... k和Ak + 1> Bk + 1。

解题思路

采用暴力求解法。将权值存在一个邻接表chains中,从根节点00开始访问,递归遍历到每个节点的孩子节点

getSum(){
   if(该节点为叶子节点,并且路径权值==S){
        路径加入结果集res中
    }else{
      for(遍历节点start的每一个孩子节点chain[start][i]]);){
          if(现有路径权值和加上节点chain[start][i]]);的权值小于S){
                将结点chain[start][i]放入路径中。
                路径权值+=chain[start][i]])的权值
                getSum(...,....,)
                将结点chain[start][i]路径中移除。
        }
      }
  }
}

得到结果res之后,定义排序规则,将res排序。

int cmp(vector<int> a,vector<int> b){
   for(int i = 0;i<a.size()&&i<b.size();i++){
       if(a[i]>b[i])return a[i]>b[i];
   }
   return a.size()<b.size();
}
sort(res.begin(),res.end(),cmp);

最后按输出格式输出res。

实现代码:

#include<stdio.h>
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> vec(100);//存储每个节点的weight
int cmp(vector<int> a,vector<int> b){
   for(int i = 0;i<a.size()&&i<b.size();i++){
       if(a[i]>b[i])return a[i]>b[i];
   }
   return a.size()<b.size();
}
void getsum(vector<vector<int> >&chain, int start,int csum,int sum,vector<vector<int> >&output, vector<int>res) {
   if (csum == sum&&chain[start].size()==0) {//判断是否为叶子节点,权值是否等于Sum。
       output.push_back(res);
       return;
   }else{
       for (int i = 0; i < chain[start].size(); i++) {
           if(csum + vec[chain[start][i]]<=sum){//只有加上该节点小于sum时方有可能为所求路径
                res.push_back(vec[chain[start][i]]);
                getsum(chain, chain[start][i], csum + vec[chain[start][i]], sum,output, res);
                res.pop_back();
           }
       }
   }

}

int main() {
   int n,m,sum,k,id;//节点个数
   cin >> n >> m >> sum;
   for (int i = 0; i < n; i++) {
       cin >> vec[i];//得到每个节点的weight
   }
   int t;
   vector<vector<int> >chains(n);//存储树
   for (int i = 0; i < m; i++) {
       cin >> id >> k;
       for (int j = 0; j < k; j++) {//得到每个父节点的k个子节点
           cin >> t;
           chains[id].push_back(t);
       }
   }
   vector<vector<int> >output;//结果
   vector<int> path;//临时路径
   res.push_back(vec[0]);
   getsum(chains, 0, vec[0], sum,output,path);
      sort(output.begin(),output.end(),cmp); //得到正确的输出顺序
   for(int i =0;i<output.size();i++){
       for(int j = 0;j<output[i].size();j++){
           cout<<output[i][j];
           if(j<output[i].size()-1)cout<<" ";
       }
       cout<<endl;
   }

   return 0;
}

我是6J,爱吃爱玩爱笑的非典型程序~~~媛

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • mean to add the formatted="false" attribute?.[ 46% 47325/...
    ProZoom阅读 7,588评论 0 3
  • 连翘花语:春天的守望者 绘画过程: 纸张:普通空白笔记本 彩铅:辉柏嘉72色水溶性彩铅 1. 貌似停了一周多没画,...
    梦扬云子阅读 5,190评论 6 13
  • 今年春节,玩转微信各种小程序,其中一张显示,2018年蛋妈适合学习与写作。嘿嘿,这不,午睡期间,猛然想起蛋哥与我的...
    耳东小猫熊阅读 5,897评论 0 1
  • 4.1正是这偶然的一瞥,成为这场半世纪后仍未结束的惊天动地的爱情的源头。 4.2但他是认真的,总有一天,他会过来交...
    SummarySu阅读 2,958评论 0 0
  • 概述: 在开发过程中,通用的功能我们通常会定义一些接口,但随着项目越做越大,时间越来越久,接口就越来越多,忘的就越...
    温木先生阅读 14,630评论 1 4