练习赛补题--------E - Treasure Hunt I DP

Akiba is a dangerous country since a bloodsucker living there. Sometimes the bloodsucker will appear and kill everyone who isn’t at his hometown. One day, a brave person named CC finds a treasure map, and he wants to get as much as possible.

Akiba consists of n towns and n-1 roads. There is a way from each town to any other. Each town contains some treasure values Vi. CC starts from town k(his hometown), at day 0. After m days, the bloodsucker will appear and CC would be killed if he hasn’t been back yet, it means CC has m days for hunting the treasure at most. It takes CC Ti days to move from one town to another neighbour town.(Two towns called neighbour if they are the endpoint of one road.) You can assume CC will get the treasure immediately as he arrives at that town. CC wants to obtain as much value as possible, keeping him alive at the same time.

Input
There are multiple cases, about 50 cases.
The first line of each case contains an integer n, indicating there are n towns.
The following line describe the treasure’s value in each town. “V1 V2 … Vn”. Vi is the value of the treasure in ith town. Each value is separated by one blank.
The next n-1 lines describe the n-1 roads in Akiba. “i j Ti” Means the ith town and the jth town are endpoints of that road. It takes Ti days to get through this road.
The last line has two integer k and m as described above.

1<=n<=100, 0<=Vi<=1000 , 1<=Ti<=10
1<=k<=n, 1<=m<=200
All the inputs are integers.

Output
Just output the max value CC can get, and you should keep CC alive after m days.

Sample Input
2
1 3
1 2 1
1 2
2
1 3
2 1 1
2 1
2
3 3
1 2 1
2 5
Sample Output
4
3
6
Hint
Sample 1: CC can go to town 2 and return at day 2.
Sample 2: CC can’t come back within 1 day. So he can only take the treasure in his hometown.
Sample 3: CC only need 2 days to collect all the treasure.

树上DP;
n个点,n-1条边,一共有m天,每条边都会有消耗的天数,又因为每次出去,必须在m天内回到原先出发的地方,因为一棵树的原因,那么它一定会原路返回,所以回程可以不需要考虑,将m天缩为m / 2天即可,
dp[u][j]代表从u点出发,有最大代价为j所能获得的最大价值;
边界值处理:dp[u][0] = val[u];(价值数)
dp[u][j] = max(dp[u][j] , dp[u][j - w[u][v] - k] + dp[v][k]);
v是u的邻接点。

#include
using namespace std;
typedef long long LL;
const int inf = 0x3f3f3f3f;
#define mp make_pair
#define pb push_back
#define fi first
#define se second

vectorve[105];
int mk[105][105];
int dp[105][205];
int val[105];
int n,k,m;
int vis[105];

void dfs(int u)
{
    vis[u] = 1;
    dp[u][0] = val[u];
    int len = ve[u].size();
    for(int i = 0;i < len;++i){
        int v = ve[u][i];
        if(vis[v] == 0){
            dfs(v);
            for(int j = m;j >= 0;--j){
                for(int x = 0;x <= j - mk[u][v];++x){
                    dp[u][j] = max(dp[u][j],dp[u][j - mk[u][v] - x] + dp[v][x]);
                }
            }
        }
    }
}

int main()
{
    while(~scanf("%d",&n)){
        for(int i = 1;i <= n;++i){
            ve[i].clear();
        }
        memset(dp,0,sizeof(dp));
        for(int i = 1;i <= n;++i){
            scanf("%d",&val[i]);
            dp[i][0] = val[i];
        }
        for(int i = 0;i < n - 1;++i){
            int u,v,w;
            scanf("%d %d %d",&u,&v,&w);
            ve[u].pb(v);
            ve[v].pb(u);
            mk[u][v] = mk[v][u] = w;
        }
        scanf("%d %d",&k,&m);
        m /= 2;
        memset(vis,0,sizeof(vis));
        dfs(k);
        int ans = -1;
        for(int i = 0;i <= m;++i){
            ans = max(ans,dp[k][i]);
        }
        printf("%d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(ACM__动态规划)