dp优化专辑 A - Starship Troopers [dp+背包]

最新推荐文章于 2022-04-20 21:58:55 发布

Blithe

最新推荐文章于 2022-04-20 21:58:55 发布

阅读量512

点赞数

分类专栏：暑假集训文章标签：优化 struct bugs 存储

本文链接：https://blog.csdn.net/lmyclever/article/details/6693529

版权

暑假集训专栏收录该内容

60 篇文章 0 订阅

订阅专栏

题意：

有n个房间组成一棵树，你有m个士兵，从1号房间开始让士兵向相邻的房间出发，每个房间里有一个brain，每一个士兵可以消灭20个bugs，求可以获得的最大brain

分析：

树形01背包，树用邻接表存储，深度优先遍历该树~dp[i][j]表示在i结点，放j个人所获得的最大值

//AC CODE：

#include<iostream>
#include<cmath>
#include<algorithm>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<string>
#include<map>

using namespace std;

const int maxn=220;
struct Node
{
    int number,p;
} node[maxn];

int n,m;
bool vis[maxn];
int dp[maxn][maxn];//dp[i][j]表示在i结点，放j个人所获得的最大值
int adj[maxn][maxn];//邻接表

void dfs(int k)
{
    vis[k]=true;
    int num=(node[k].number+19)/20;
    for(int i=num; i<=m; i++)
        dp[k][i]=node[k].p;
    for(int i=1; i<=adj[k][0]; i++)
    {
        int u=adj[k][i];
        if (vis[u])//别忘了剪枝
            continue;//
        dfs(u);
        //关键部分 01 package
        for(int j=m; j>=num; j--)
            for(int l=1; l+j<=m; l++)
                dp[k][j+l]=max(dp[k][j+l],dp[k][j]+dp[u][l]);
    }
}

int main()
{
    while(scanf("%d%d",&n,&m)&&(n!=-1&&m!=-1))
    {
        memset(vis,false,sizeof(vis));
        memset(dp,0,sizeof(dp));
        memset(adj,0,sizeof(adj));
        for(int i=1; i<=n; i++)
            scanf("%d%d",&node[i].number,&node[i].p);
        int b,e;
        for(int i=1; i<n; i++)//用邻接表存树
        {
            scanf("%d%d",&b,&e);
            adj[b][0]++;
            adj[b][adj[b][0]]=e;
            adj[e][0]++;
            adj[e][adj[e][0]]=b;
        }
        if(m==0)
            printf("0\n");
        else
        {
            dfs(1);
            printf("%d\n",dp[1][m]);
        }
    }
}