【PAT_1053 】 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);
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)return;
	if (csum == sum) {
	  if(chain[start].size()!=0)return;
		output.push_back(res);
		return;
	}else{
	
        for (int i = 0; i < chain[start].size(); i++) {
            if(csum + vec[chain[start][i]]<=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];
	}
	int t;
	vector<vector<int> >chains(n);
	for (int i = 0; i < m; i++) {
		cin >> id >> k;
		for (int j = 0; j < k; j++) {
			cin >> t;
			chains[id].push_back(t);
		}
	}
	vector<vector<int> >output;
	vector<int> res;
	res.push_back(vec[0]);
	getsum(chains, 0, vec[0], sum,output,res);
    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,爱吃爱玩爱笑的非典型程序~~~媛