POJ 2486

POJ 2486

对于一个弱菜来说，期末考试是鬼门关，经过一个月的预习，终于完成了大学生涯第一次不挂科的经历，也要回到ACM中被虐的征途了。

题意：给出一颗N个节点的树，根节点为1，然后给出每个节点苹果的数目，问在走K步之内最多能吃掉多少苹果，可以走回头路，回头路这个太恶心了，断断续续想了几个星期，今天试着交一次，居然一次AC了，惊喜。

思路：

状态DP[ I ][ J ][ K ],代表以J为根的是否能走回头路的走了K步吃到的最多苹果数。

对于每个J，对他所有的儿子都做一次相当于背包的操作，要注意区分回头/不回头的细节，不回头只能有一次，而回头可以多次，处理下细节，OK~

#include <cstdio>
#include <cstring>
#include <cmath>
#include <queue>
#include <stack>
#include <vector>
#include <string>
#include <map>
#include <set>
#include <sstream>
#include <iostream>
#include <algorithm>
#include<cstdlib>

using namespace std;

#define N 250
#define L(x) x<<1
#define R(x) x<<1|1
#define M(x,y) (x + y)>>1
#define MOD 1000000007
#define MODD 1000000006
#define inf 0x7fffffff
#define llinf 0x7fffffffffffffff
#define LL long long

struct st
{
	int u,v;
	int next;
}s[N*2];

int cnt;
int visit[N];
int dp[2][N][N];          //0回头，1不回头
int mark[N];
int num[N];
int n,m;

void addedge(int a,int b)
{
	s[cnt].u = a;
	s[cnt].v = b;
	s[cnt].next = visit[a];
	visit[a] = cnt++;
	s[cnt].u = b;
	s[cnt].v = a;
	s[cnt].next = visit[b];
	visit[b] = cnt++;
}

void dfs(int x)
{
	mark[x] = 1;
	dp[0][x][0] = num[x];
	dp[1][x][0] = num[x];
	for(int i = visit[x];i != -1;i = s[i].next)
	{
		if(mark[s[i].v] == 1 )
			continue;
		dfs(s[i].v);
		for(int j = m;j > 0;j--)
		{
			for(int k = 0;k <= j - 2;k++)
			{
				dp[1][x][j] = max(dp[1][x][j], dp[0][s[i].v][k] + dp[1][x][j - k - 2]);
			}
			for(int k = 0;k <= j - 1;k++)
			{
				dp[1][x][j] = max(dp[1][x][j], dp[1][s[i].v][k] + dp[0][x][j - k - 1]);
			}
			for(int k = 0;k <= j - 2;k++)
			{
				dp[0][x][j] = max(dp[0][x][j], dp[0][s[i].v][k] + dp[0][x][j - k - 2]);
			}
		}
	}
}

int main()
{
	int i,j,k,l;
	
	while(scanf("%d%d",&n,&m) != EOF)
	{
		cnt = 0;
		memset(visit,-1,sizeof(visit));
		memset(num,0,sizeof(num));
		memset(dp,0,sizeof(dp));
		for(i = 1;i <= n;i++)
		{
			scanf("%d",&num[i]);
		}
		for(i = 1;i < n;i++)
		{
			scanf("%d%d",&j,&k);
			addedge(j,k);
		}
		memset(mark,0,sizeof(mark));
		dfs(1);
		int ans = -1;
		for(i = 0;i <= m;i++)
		{
			ans = max(ans,dp[0][1][i]);
			ans = max(ans,dp[1][1][i]);
		}
		printf("%d\n",ans);
	}
	return 0;
}