1053. Path of Equal Weight

最新推荐文章于 2022-04-13 00:26:50 发布

原创最新推荐文章于 2022-04-13 00:26:50 发布 · 801 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#acm #pat #树 #路径 #dfs

PAT 专栏收录该内容

70 篇文章

订阅专栏

本文提供了一种使用C++实现的PAT-A 1053题目的解决方案，通过邻接表表示树结构，并利用DFS深度优先搜索算法找出所有路径中节点值之和等于给定值S的路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：http://pat.zju.edu.cn/contests/pat-a-practise/1053

同一类型：路径长度、保存与给定值相同的路径长度、记录一定深度的路径、记录节点号/值。

// 采用邻接表表示树
// 用vector存储路径
// 路径不止1一条，需用vector< vector<int> >存储所有路径

// 输入时建树
// 得到一个节点的所有子节点后，进行排序：按权值
// dfs遍历，叶节点处判断。返回上一层时，去掉path中的权值



#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <ctype.h>
#include <limits.h>

#include <iostream>
#include <algorithm>
#include <vector>
#include <string>
#include <queue>
#include <stack>
#include <map>
#include <set>

#define SIZE 100+10

using namespace std;


struct Node
{
	int key;
	vector<int> child;
};

int N, M, S;
vector<Node> Tree;
vector<int> path;
vector<vector<int> > ans;

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

void Input()
{
	scanf("%d%d%d", &N, &M, &S);
	int i;
	for(i = 0; i < N; i++)
	{
		Node t;
		scanf("%d", &t.key);
		Tree.push_back(t);
	}

	// input and creating tree
	for(i = 0; i < M; i++)
	{
		int id, num;
		scanf("%d%d", &id, &num);
		while(num-- > 0)
		{
			int child_id;
			scanf("%d", &child_id);
			Tree[id].child.push_back(child_id);
		}

		//sorting
		sort(Tree[id].child.begin(), Tree[id].child.end(), cmp);
	}
	return ;
}

void dfs(int id, int sum)
{
	path.push_back(Tree[id].key);
	sum = sum + Tree[id].key;

	// leaf
	if(Tree[id].child.size() == 0)
	{
		if(sum == S)
		{
			ans.push_back(path);	
			return ;
		}// 得到一条路径
		else
		{

			return ;
		}
	}
	else
	{
		if(sum < S)
		{
			for(int j=0; j<Tree[id].child.size(); j++)
			{
				int next = Tree[id].child[j];
				dfs(next, sum);
				path.pop_back();//当前节点退出路径
			}
			return ;//所有子节点遍历完成，返回
		}
		else
		{
			return ;//非叶节点的和已经大于等于S，返回
		}
	}
}

// 构造树后输出结果，验证是否正确
void Output()
{
	for(int i=0; i < Tree.size(); i++)
	{
		printf("%02d:", i);
		for(int j=0; j<Tree[i].child.size(); j++)
		{
			printf("%d ", Tree[i].child[j]);
		}
		printf("\n");
	}
}

int main()
{
	#ifdef ONLINE_JUDGE
	#else
		freopen("E:\\in.txt", "r", stdin);
	#endif

	Input();
//	Output();

	int sum = 0;
	dfs(0, sum);

	// output
	for(int i=0; i < ans.size(); i++)
	{
		for(int j=0; j<ans[i].size()-1; j++)
		{
			printf("%d ", ans[i][j]);
		}
		printf("%d\n", ans[i][ans[i].size()-1]);
	}
	return 0;
}