hdu 1011 Starship Troopers(树形DP入门)

Starship Troopers

Time Limit: 10000/5000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 17498    Accepted Submission(s): 4644


Problem Description
You, the leader of Starship Troopers, are sent to destroy a base of the bugs. The base is built underground. It is actually a huge cavern, which consists of many rooms connected with tunnels. Each room is occupied by some bugs, and their brains hide in some of the rooms. Scientists have just developed a new weapon and want to experiment it on some brains. Your task is to destroy the whole base, and capture as many brains as possible.

To kill all the bugs is always easier than to capture their brains. A map is drawn for you, with all the rooms marked by the amount of bugs inside, and the possibility of containing a brain. The cavern's structure is like a tree in such a way that there is one unique path leading to each room from the entrance. To finish the battle as soon as possible, you do not want to wait for the troopers to clear a room before advancing to the next one, instead you have to leave some troopers at each room passed to fight all the bugs inside. The troopers never re-enter a room where they have visited before.

A starship trooper can fight against 20 bugs. Since you do not have enough troopers, you can only take some of the rooms and let the nerve gas do the rest of the job. At the mean time, you should maximize the possibility of capturing a brain. To simplify the problem, just maximize the sum of all the possibilities of containing brains for the taken rooms. Making such a plan is a difficult job. You need the help of a computer.
 
Input
The input contains several test cases. The first line of each test case contains two integers N (0 < N <= 100) and M (0 <= M <= 100), which are the number of rooms in the cavern and the number of starship troopers you have, respectively. The following N lines give the description of the rooms. Each line contains two non-negative integers -- the amount of bugs inside and the possibility of containing a brain, respectively. The next N - 1 lines give the description of tunnels. Each tunnel is described by two integers, which are the indices of the two rooms it connects. Rooms are numbered from 1 and room 1 is the entrance to the cavern.

The last test case is followed by two -1's.
 
Output
For each test case, print on a single line the maximum sum of all the possibilities of containing brains for the taken rooms.
 
Sample Input
5 10 50 10 40 10 40 20 65 30 70 30 1 2 1 3 2 4 2 5 1 1 20 7 -1 -1
 
Sample Output
50 7
 
Author
XU, Chuan
 
Source
 
Recommend
JGShining   |   We have carefully selected several similar problems for you:  1561 2415 1114 2546 2159 
 
/*
这个必须要用无向图,重判的时候用visit数组标记
*/
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<vector>
#define N 110
using namespace std;
int n,m;
vector<int> v[N];//用来存放数的
int dp[N][N];//dp[i][j]表示在第i个结点的时候,用j个士兵能获得多少幽灵
int f[N];//用来找树根的
bool visit[N];//记录当前结点走没走过
struct node
{
    int p,q;//妖怪的数量,想要的有幽灵的数量
}fr[N];
void dfs(int root)
{
    visit[root]=true;
    int num=(fr[root].p+19)/20;//当前房间最少需要
    for(int i=num;i<=m;i++)
            dp[root][i]=fr[root].q;//只要是超过了num个士兵都是q个人
    //cout<<num<<endl;
    for(int i=0;i<v[root].size();i++)
    {
        int next=v[root][i];//下一个房间
        //cout<<"next="<<next<<endl;
        //cout<<"visit[next]="<<visit[next]<<endl;
        if(visit[next])
        {
            //cout<<"visit[next]="<<visit[next]<<endl;
            continue;//下一步走过了就跳过
        }    
        dfs(next);
        //cout<<"ok"<<endl;
        //对dp数组进行初始化
        for(int j=m;j>=num;j--)//背包枚举(01背包嘛从后往前枚举)
        {
            for(int k=1;k+j<=m;k++)
            {
                if(dp[next][k])
                    dp[root][j+k]=max(dp[root][j+k],dp[root][j]+dp[next][k]);//用j+k些士兵在当前房间,当前房间用j个士兵下一个房间用k个士兵,这两者哪一个大
            }    
        }
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        if(n==-1&&m==-1)
            break;
        memset(visit,false,sizeof visit);
        memset(dp,0,sizeof dp);
        v[0].clear();
        for(int i=1;i<=n;i++)
        {
            scanf("%d%d",&fr[i].p,&fr[i].q);
            //cout<<fr[i].p<<" "<<fr[i].q<<endl;
            //f[i]=i;
            v[i].clear();
        }
        v[n+1].clear();
        int a,b;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            //cout<<a<<" "<<b<<endl;
            v[a].push_back(b);
            v[b].push_back(a);
            f[b]=a;
        }
        //for(int i=1;i<=n;i++)
        //    cout<<"i="<<i<<" "<<v[i].size()<<endl;
        if(m==0)
        {
            printf("0
");
        }
        else
        {
            //int root=1;
            //while(root!=f[root]) root=f[root];//找根
            //cout<<"root="<<root<<endl;
            dfs(1);
            printf("%d
",dp[1][m]);

        }
    }
    return 0;   
}
原文地址:https://www.cnblogs.com/wuwangchuxin0924/p/5781943.html