hdu 4044 GeoDefense

hdu 4044 GeoDefense

这个题目足足调试了我一个晚上呀！ 看来dp弱的要死了，到了不练习不行的地步了

题意： 给出一棵树，每个树根节点为1，其实就是求怎么大炮才能使根节点到任意一个叶子节点的最小值最大，并且总花费不超过m

有几个tick就是price 可能为0，这就给背包带来了很容易忽视的错误，其实dp状态转移还是很好想的，具体看代码吧

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>

using namespace std;
const int maxn=1010;
const int inf=0x3fffffff;
vector<int> g[maxn];
int dp[maxn][210],m,tmp[maxn];
vector<pair<int,int> > tower[maxn];


void dfs(int u,int fa)
{
    int sz=g[u].size();
    bool ok=0;
    if(sz==1&&g[u][0]==fa) ok=1;  // ok=1 这是叶子节点

    if(!ok)  // 非叶子节点
    {
       for(int i=0;i<=m;i++) dp[u][i]=inf; // 当第一次是要赋值为inf
       for(int rr=0;rr<sz;rr++)
       {
          int v=g[u][rr];
          if(v==fa) continue;
          dfs(v,u);
           for(int i=m;i>=0;i--)
           {
                int t=0;       // 不能写成dp[u][i]=0,应为j可能为0
                for(int j=0;j<=i;j++)
                  t=max(t,min(dp[u][i-j],dp[v][j]));
               dp[u][i]=t;
           }
       }
    }

    for(int i=0;i<=m;i++) tmp[i]=dp[u][i];  // 保存先前的dp值

    int k=tower[u].size();
    for(int i=m;i>=0;i--)
    for(int j=0;j<k;j++)
       if(tower[u][j].first<=i)
          dp[u][i]=max(dp[u][i],tmp[i-tower[u][j].first]+tower[u][j].second);
}
int main()
{
    int T,a,b,k,n;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d",&n);
        for(int i=1;i<=n;i++) tower[i].clear(),g[i].clear();
        for(int i=1;i<n;i++)
        {
            scanf("%d%d",&a,&b);
            g[a].push_back(b);
            g[b].push_back(a);
        }
        scanf("%d",&m);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&k);
            for(int j=0;j<k;j++)
            {
               scanf("%d%d",&a,&b);
               tower[i].push_back(make_pair(a,b));
            }
        }
        memset(dp,0,sizeof(dp));
        dfs(1,-1);
        printf("%d\n",dp[1][m]);
    }
    return 0;
}