HDU 4276 树形dp+spfa+分组背包

<span style="font-family: Arial, Helvetica, sans-serif; background-color: rgb(255, 255, 255);"><span style="font-size:14px;">题意：给你n个点，n-1条边构成树，每条边有边树，每个点有点权（表示走每条边的时间），问在时间T从点1走到点n，能够得到最多的点权有多少。</span></span>
首先我们至少要保证1到n的最短路在时间内，此时用spfa，其次，在最优的情况下，最短路径上的边只会经过一次，而其他边会经过两次，对于最短路上的边，spfa走的时候记录权值，然后赋值为0就可以了。对于其他边就可以跑树形dp了。

#include <cmath>
#include <queue>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>
#define LL long long
#define INF 0x3f3f3f3f
using namespace std;
const int maxn = 5e4+7;
int dp[111][555],head[111],v[111],d[111],t[111],num,n,kk;
struct list1 {int u,v,w,next;}a[222];
void add(int u,int v,int w) {a[num]={u,v,w,head[u]};head[u]=num++;}
void spfa()
{
int i;
int pre[111];
memset(pre,-1,sizeof(pre));
memset(v,0,sizeof(v));
memset(d,INF,sizeof(d));
queue<int>q;
q.push(1);v[1]=1;d[1]=0;
while(!q.empty())
{
int u=q.front();q.pop();v[u]=0;
for(i=head[u];~i;i=a[i].next)
{
int u=a[i].u;
if(d[u]+a[i].w<d[a[i].v])
{
pre[a[i].v]=i;
d[a[i].v]=d[u]+a[i].w;
if(!v[a[i].v]) q.push(a[i].v),v[a[i].v]=1;
}
}
}

4000
for(i=n;i!=1;i=a[pre[i]].u)
{
a[pre[i]].w=0;
a[pre[i]^1].w=0;
}
}
void dfs(int x,int fa)
{
for(int i=head[x];~i;i=a[i].next)
{
int u=a[i].v,w=2*a[i].w;
if(u==fa) continue;
dfs(u,x);
for(int j=kk;j>=w;j--)
for(int k=0;k<=j-w;k++)
dp[x][j]=max(dp[x][j],dp[x][j-k-w]+dp[u][k]);
}
}
int main()
{
int i,x,y,z;
while(~scanf("%d%d",&n,&kk))
{
memset(dp,0,sizeof(dp));
num=0;memset(head,-1,sizeof(head));
for(i=1;i<n;i++)
{
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);add(y,x,z);
}
for(i=1;i<=n;i++)
{
scanf("%d",&dp[i][0]);
for(int j=0;j<=kk;j++)
dp[i][j]=dp[i][0];
}
if(n==1) {printf("%d\n",dp[1][0]);continue;}
spfa();kk-=d
;
if(kk<0) printf("Human beings die in pursuit of wealth, and birds die in pursuit of food!\n");
else dfs(1,0),printf("%d\n",dp[1][kk]);
}
return 0;
}