对于一个弱菜来说,期末考试是鬼门关,经过一个月的预习,终于完成了大学生涯第一次不挂科的经历,也要回到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;
}