Poj 1655 Balancing Act (DP_树形DP)

题目链接:http://poj.org/problem?id=1655

题目大意:给定一棵n棵节点的树,删去某个节点后剩下来的分支中肯定会有最大节点数,求所有节点的最大分支节点数的最小值。

解题思路:树形DP.深搜两次,第一次深搜记录从当前节点的子孙节点总数(包括自己),第一次算预处理,复杂度为O(N),第二次利用第一次的结果找各分支的最大节点数,分支有两种情况,一种是孩子的分支,一种当前节点到父亲节点的那条分支(总数为N-dp[cur]),这样再算N次即可得解。


测试数据:

1
7
2 6
1 2
1 4
4 5
3 7
3 1


代码:
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define MAX 110000
#define max(a,b) (a)>(b)?(a):(b)


struct node {

	int v;
	node *next;
}*head[MAX],tree[MAX];
int n,m,ptr,dp[MAX],ansval,ansi,cnt;


void Initial() {

	cnt = ptr = 0;
	ansval = ansi = 2147483647;
	memset(dp,0,sizeof(dp));
	memset(head,NULL,sizeof(head));
}
void AddEdge(int x,int y) {

	tree[ptr].v = y;
	tree[ptr].next = head[x],head[x] = &tree[ptr++];
	tree[ptr].v = x;
	tree[ptr].next = head[y],head[y] = &tree[ptr++];
}
void Dfs_Ini(int s,int pa) {

	dp[s] = 1;
	node *p = head[s];
	while (p != NULL) {

		if (p->v != pa) {

			Dfs_Ini(p->v,s);
			dp[s] += dp[p->v];
		}
		p = p->next;
	}
}
void Dfs_Solve(int son,int pa) {

	int i,j,tp,tot = 0;
	node *p = head[son];
	

	while (p != NULL) {

		if (p->v != pa) {

			Dfs_Solve(p->v,son);
			tp = dp[p->v];
			tot = max(tot,tp);
		}
		p = p->next;
	}
	if (tot < n - dp[son])
		tot = n - dp[son];
	if (tot < ansval)
		ansval = tot,ansi = son;
}


int main()
{
	int i,j,k,a,b,t;


	scanf("%d",&t);
	while (t--) {
	
		scanf("%d",&n);
		Initial();
		for (i = 1; i < n; ++i) {

			scanf("%d%d",&a,&b);
			AddEdge(a,b);
		}


		Dfs_Ini(1,0);			//第一次深搜,记录当前节点的子孙总数
		Dfs_Solve(1,0);			//更新答案
		printf("%d %d\n",ansi,ansval);
	}
}


本文ZeroClock原创,但可以转载,因为我们是兄弟。

你可能感兴趣的:(struct,tree,测试,null,ini)