HDU 1655 Balancing Act

最新推荐文章于 2021-04-20 10:56:04 发布

little_胖

最新推荐文章于 2021-04-20 10:56:04 发布

阅读量735

点赞数

CC 4.0 BY-SA版权

分类专栏： DP

本文链接：https://blog.youkuaiyun.com/wangjie_wang/article/details/9257109

DP 专栏收录该内容

66 篇文章

订阅专栏

本文介绍了一种基于树状结构的动态规划算法实现，并通过一个具体示例代码展示了如何求解树的最大子树问题。该算法通过递归方式计算每个节点的子树最大值和子树节点总数。

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

也是个简单树状DP，和POJ3107的写法基本一致。

#include <cstdio>
#include <cstring>
#include <string>
#include <iostream>
#include <queue>
#include <algorithm>
#include <map>
#include <vector>
#include <cmath>
#include <stack>
using namespace std;
#define ll long long
#define int64 __int64
#define M 100005
#define N 1005
#define inf 1000010
#define mod 1000000007

struct node
{
	int son_max;
	int son_sum;
	int pos;
}tp[M];
struct Node
{
	int ev , next;
}tree[M];
int n , tot , head[M];

void Add(int s , int e)
{
	tree[tot].ev = e;
	tree[tot].next = head[s];
	head[s] = tot++;
}

void Dfs(int s , int fa)
{
	int i;
	tp[s].son_max = 0;
	tp[s].son_sum = 1;
	tp[s].pos = s;
	for (i = head[s] ; i != -1 ; i = tree[i].next)
	{
		int v = tree[i].ev;
		if (v == fa)continue;
		Dfs(v,s);
		tp[s].son_max = max(tp[s].son_max , tp[v].son_sum);
		tp[s].son_sum += tp[v].son_sum;
	}
	tp[s].son_max = max(tp[s].son_max , n-tp[s].son_sum);
}

bool cmp(node a , node b)
{
	if (a.son_max < b.son_max)return true;
	if (a.son_max > b.son_max)return false;
	if (a.pos < b.pos)return true;
	return false;
}

int main()
{
	int i , t;
	scanf("%d",&t);
	while (t--)
	{
		tot = 0;
		memset(head , -1 , sizeof head);
		scanf("%d",&n);
		for (i = 1 ; i < n ; i++)
		{
			int s , e;
			scanf ("%d%d",&s,&e);
			Add(s,e);
			Add(e,s);
		}
		Dfs(1,0);
		sort(tp+1 , tp+n+1 , cmp);
		printf("%d %d\n",tp[1].pos,tp[1].son_max);
	}
	return 0;
}