【PAT甲级】1053 Path of Equal Weight (30分)_pat甲级 path of equal weight(30分)c语言-优快云博客

解题过程的小记录，如有错误欢迎指出。

难度：四星（难点在与记录结点的路径并进行输出）

小导航~

题目分析
- - 注意点
我的解题过程
dalao的代码
- - 借鉴点

题目分析

每个结点都有自己的重量，不妨设从root到叶子的重量之和为总重量，要求找出总总量与给定值相等的叶子结点，并输出从root到它的重量

注意点

要求输出的是重量而不是下标
找出后的path要进行排序后输出

我的解题过程

思路

设置node结构体，有重量weight、孩子child、从root到自身的path属性，并开一个大于最大结点数的数组Node
输入数据weight，和树的结构（完整每个结点的child变量）
写DFS函数，参数设置为当前下标index和除此结点外的path之前的总重量weight，如果结点是叶子结点且加上当前index的weight与给定的重量值相等，说明此叶子为要找的，如果不是叶子结点，则扫描它的孩子，并进一步DFS（这里要注意path的赋值问题，详见代码）
找到所有符合的叶子结点把他们存入ans数组，写一个comp函数进行排序
输出

bug

path的赋值问题，因为刚开始只在每一层写了push，这样的话就只会push进当前的index，而此之前的path不会赋值
打印问题，打印的下标写错成j了，其实应该是每个ans的path值

代码

#include<iostream>
#include<vector>
#include<algorithm>

using namespace std;

struct node {
	int weight;
	vector<int> child;
	vector<int> path;
}Node[105];

vector<node> ans;
int answeight;

void DFS(int index, int weight) {//index是当前的下标，weight是此条path之前的重量和（不包含index的重量
	if (Node[index].child.size() == 0) {
		Node[index].path.push_back(index);
		if (weight + Node[index].weight == answeight) {
			ans.push_back(Node[index]);
		}
		return;
	}
	for (int i = 0; i < Node[index].child.size(); i++) {
		int child = Node[index].child[i];
		Node[child].path = Node[index].path;//**要先把这之前的祖先复制下来，不然会只有一个被计入
		Node[child].path.push_back(index);//赋值完了后，把当前的赋值给它的孩子
		if (weight + Node[index].weight + Node[child].weight <= answeight)
			DFS(child, weight + Node[index].weight);
	}
}

int comp(node n1, node n2) {
	int i = 0;
	for (i; i < n1.path.size() && i < n2.path.size(); i++) {
		int w1 = Node[n1.path[i]].weight;
		int w2 = Node[n2.path[i]].weight;
		if (w1 > w2) return 1;
		else if (w1 < w2) return 0;
	}
	if (n1.path.size() > n2.path.size()) return 1;
	return 0;
}

int main()
{
	int n, m;
	scanf("%d %d %d", &n, &m, &answeight);
	for (int i = 0; i < n; i++) {
		scanf("%d", &Node[i].weight);
	}
	for (int i = 0; i < m; i++) {
		int parent, k;
		scanf("%d %d", &parent, &k);
		for (int j = 0; j < k; j++) {
			int index;
			scanf("%d", &index);
			Node[parent].child.push_back(index);
		}
	}
	DFS(0, 0);
	sort(ans.begin(), ans.end(), comp);
	for (int i = 0; i < ans.size(); i++) {
		for (int j = 0; j < ans[i].path.size(); j++) {
			int index = ans[i].path[j];//**找到path的下标
			if (j != 0) printf(" ");
			printf("%d", Node[index].weight);
		}
		printf("\n");
	}
    return 0;
}