zoj 3626 Treasure Hunt I

题目链接：http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3626

题目大意：

给定一个n（1~100）个结点棵树，边和点均有权值（点权0~100，边权1~10）。

从结点k（根）出发，给定一个限定值m。
设值sv为经过的点的权值和，设se为经过的边的权值和。

点权值只加一次，边权值经过一次加一次。

问se<=m时最大的sv值。

题目思路：

树形dp。

dp[u][j]:以u为根的子树，用j天能获得的最大值

dp[u][j]=max(dp[u][j],dp[u][j-dis[u][v]-i]+dp[v][i]),

v为u的子节点，dis[u][v]:u-v边的权值

代码：

#include <stdlib.h>
#include <string.h>
#include <stdio.h>
#include <ctype.h>
#include <math.h>
#include <stack>
#include <queue>
#include <map>
#include <set>
#include <vector>
#include <string>
#include <iostream>
#include <algorithm>
using namespace std;

#define ll __int64
#define ls rt<<1
#define rs ls|1
#define lson l,mid,ls
#define rson mid+1,r,rs
#define middle (l+r)>>1
#define clr_all(x,c) memset(x,c,sizeof(x))
#define clr(x,c,n) memset(x,c,sizeof(x[0])*(n+1))
#define eps (1e-8)
#define MOD 1000000007
#define INF 0x3f3f3f3f
#define PI (acos(-1.0))
#pragma comment(linker, "/STACK:102400000,102400000")
template <class T> T _max(T x,T y){return x>y? x:y;}
template <class T> T _min(T x,T y){return x<y? x:y;}
template <class T> T _abs(T x){return (x < 0)? -x:x;}
template <class T> T _mod(T x,T y){return (x > 0)? x%y:((x%y)+y)%y;}
template <class T> void _swap(T &x,T &y){T t=x;x=y;y=t;}
template <class T> void getmax(T& x,T y){x=(y > x)? y:x;}
template <class T> void getmin(T& x,T y){x=(x<0 || y<x)? y:x;}
int TS,cas=1;
const int M=100+5;
int n,s,m;
int v[M],dis[M][M],dp[M][M];
int fa[M],vex[M<<1],nxt[M<<1];
bool vis[M];

void dfs(int u){
	vis[u]=1;
	for(int i=fa[u];i!=-1;i=nxt[i]) if(!vis[vex[i]]){
		int v=vex[i];
		dfs(v);
		for(int j=m;j>=dis[u][v];j--)
			for(int k=0;k<=j-dis[u][v];k++)
				dp[u][j]=_max(dp[u][j],dp[u][j-dis[u][v]-k]+dp[v][k]);
	}
}

void run(){
	int i,j,k;
	clr_all(dis,-1);
	for(i=1;i<=n;i++) scanf("%d",&v[i]),fa[i]=-1,vis[i]=0;
	for(i=1;i<n;i++){
		int u,v,t;
		scanf("%d%d%d",&u,&v,&t);
		dis[u][v]=dis[v][u]=t;
		nxt[i]=fa[u],vex[i]=v,fa[u]=i;
		nxt[i+n]=fa[v],vex[i+n]=u,fa[v]=i+n;
	}
	scanf("%d%d",&s,&m);
	m/=2;
	for(i=1;i<=n;i++) for(j=0;j<=m;j++) dp[i][j]=v[i];
	dfs(s);
	printf("%d\n",dp[s][m]);
}

void preSof(){
}

int main(){
    //freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    preSof();
    //run();
    while((~scanf("%d",&n))) run();
    //for(scanf("%d",&TS);cas<=TS;cas++) run();
    return 0;
}

设值sv为经过的点的权值和，设se为经过的边的权值和。