1079 Total Sales of Supply Chain (25)（25 分）

最新推荐文章于 2024-05-05 11:48:17 发布

原创最新推荐文章于 2024-05-05 11:48:17 发布 · 484 阅读

0 ·

CC 4.0 BY-SA版权

PAT甲级真题专栏收录该内容

37 篇文章

订阅专栏

本文提供了一个使用深度优先搜索（DFS）和广度优先搜索（BFS）算法的示例，通过这两种算法遍历一棵树形结构并计算每个叶子节点的权重值之和。示例中详细展示了如何定义树节点结构、读取输入数据以及实现DFS和BFS的具体过程。

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

DFS:递归终点为叶子结点，此时计算乘积

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
const int maxn = 1e5 + 10;
struct node {
	double p, product;
	vector<int>child;
}Node[maxn];
int n; 
double p, r;
double ans = 0;
void DFS(int index)
{
	if (Node[index].child.size() == 0)
	{
		ans += Node[index].p*Node[index].product;
		return;
	}
	for (int i = 0; i < Node[index].child.size(); i++)
	{
		int child = Node[index].child[i];
		Node[child].p = Node[index].p*(1 + r);
		DFS(child);
	}
}
int main()
{
	scanf("%d%lf%lf", &n, &p, &r);
	r /= 100;
	for (int i = 0; i < n; i++)
	{
		int k;
		scanf("%d", &k);
		if (k)
		{
			while (k--)
			{
				int x;
				scanf("%d", &x);
				Node[i].child.push_back(x);
			}
		}
		else
		{
			scanf("%lf", &Node[i].product);
		}
	}
	Node[0].p = p;
	DFS(0);
	printf("%.1f", ans);
	return 0;
}

BFS:队列，分是否为叶子结点操作，不是叶子结点，计算p压进队列，是叶子结点就计算一下乘积

#include<iostream>
#include<queue>
#include<vector>
using namespace std;
const int maxn = 1e5 + 10;
struct node {
	double p, product;
	vector<int>child;
}Node[maxn];
int n; 
double p, r;
double ans = 0;
void BFS(int root)
{
	queue<int>q;
	q.push(root);
	while (!q.empty())
	{
		int top = q.front();
		q.pop();
		if (Node[top].child.size())
		{
			for (int i = 0; i < Node[top].child.size(); i++)
			{
				int child = Node[top].child[i];
				Node[child].p = Node[top].p*(1 + r);
				q.push(child);
			}
		}
		else ans += Node[top].p*Node[top].product;
	}
}
int main()
{
	scanf("%d%lf%lf", &n, &p, &r);
	r /= 100;
	for (int i = 0; i < n; i++)
	{
		int k;
		scanf("%d", &k);
		if (k)
		{
			while (k--)
			{
				int x;
				scanf("%d", &x);
				Node[i].child.push_back(x);
			}
		}
		else
		{
			scanf("%lf", &Node[i].product);
		}
	}
	Node[0].p = p;
	BFS(0);
	printf("%.1f", ans);
	return 0;
}