最大子树和

题目:

P1122 最大子树和 - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)

模板题,用于自我复习

AC代码:

#include<bits/stdc++.h>
using namespace std;

typedef long long ll;

const int N = 2e5 + 5;

vector<int>g[N];//邻接关系

int dp[N];//存放,当前节点作为根的所有子树中,收益最大的价值

void dfs(int x, int fa)
{
	for (auto& i : g[x])//遍历此节点的所有分支
	{
		if (i == fa)continue;
        //寻找当前节点的子节点
		dfs(i, x);
		if (dp[i] > 0)如果子树能带来正收益,那就加进去
			dp[x] += dp[i];
	}
}

int main()
{
	std::ios::sync_with_stdio(0);
	cin.tie(0);
	cout.tie(0);

	int n;
	cin >> n;

	for (int i = 1; i <= n; i++)
		cin >> dp[i];

	for (int i = 1; i <= n - 1; i++)
	{
		int u, v;
		cin >> u >> v;

		if (u != v)
		{
			g[u].push_back(v);
			g[v].push_back(u);
		}
	}

	dfs(1, 0);//任意一点都可以作为根,dfs(n,0)也可以
	cout << *max_element(dp + 1, dp + 1 + n) << '\n';

	return 0;
}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
要求二叉树每个节点的左右子树高度和的最大值,需要遍历每个节点并计算其左右子树的高度,再将左右子树高度相加得到该节点的左右子树高度和,最终比较所有节点的左右子树高度和,输出最大值即可。 具体实现可以使用递归的方式遍历二叉树,在递归的过程中计算每个节点的左右子树高度和,并更新最大值。递归函数可以返回当前节点的高度,以便在计算父节点的高度时使用。 以下是C语言实现的示例代码: ```c #include <stdio.h> #include <stdlib.h> typedef struct TreeNode { int val; struct TreeNode *left; struct TreeNode *right; } TreeNode; int max(int a, int b) { return a > b ? a : b; } int getHeight(TreeNode* root) { if (root == NULL) { return 0; } int leftHeight = getHeight(root->left); int rightHeight = getHeight(root->right); return max(leftHeight, rightHeight) + 1; } int maxSum(TreeNode* root, int* maxSumValue) { if (root == NULL) { return 0; } int leftSum = maxSum(root->left, maxSumValue); int rightSum = maxSum(root->right, maxSumValue); int sum = leftSum + rightSum + root->val; if (sum > *maxSumValue) { *maxSumValue = sum; } return max(leftSum, rightSum) + root->val; } int main() { TreeNode* root = (TreeNode*)malloc(sizeof(TreeNode)); root->val = 1; root->left = (TreeNode*)malloc(sizeof(TreeNode)); root->left->val = 2; root->left->left = NULL; root->left->right = NULL; root->right = (TreeNode*)malloc(sizeof(TreeNode)); root->right->val = 3; root->right->left = (TreeNode*)malloc(sizeof(TreeNode)); root->right->left->val = 4; root->right->left->left = NULL; root->right->left->right = NULL; root->right->right = (TreeNode*)malloc(sizeof(TreeNode)); root->right->right->val = 5; root->right->right->left = NULL; root->right->right->right = NULL; int maxSumValue = 0; maxSum(root, &maxSumValue); printf("Max sum of left and right subtree heights: %d\n", maxSumValue); free(root->left); free(root->right->left); free(root->right->right); free(root->right); free(root); return 0; } ``` 以上代码中,`getHeight`函数计算二叉树的高度,`maxSum`函数使用递归计算每个节点的左右子树高度和,并更新最大值。在`main`函数中,构造了一个示例二叉树,并输出了左右子树高度和的最大值。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值