[HAOI2010]软件安装

断了今生、忘了曾经 提交于 2019-11-26 15:52:06

题意简化

    有 n 个物品和容量为 m 的背包,每个物品最多有一个先决条件物品 $(即必须选了它的先决条件物品,才能选当前物品)$ ,第 i 个物品占 $W<sub>i</sub>$ 单位空间,有 $V<sup>i</sup>$ 的价值,求最大总价值 n<=100,m<=500

solution

    看到这种有先决条件且先决条件只分 (0/1) 的背包, 第一时间想到的自然是 按先决条件连边,然后跑树形背包
    

但是,对于此题而言建边之后不一定是一棵树

而是很多个联通块

甚至还可能会有环!!!

对于有环的情况又该怎么处理呢?
若是有环,则显然环上的点都有其必要的先决条件,也就是说要把整个环一起放入背包,所以我们可以把这个环看做是一个整体
也就是用Tarjan缩点就好

而对于每个连通块,我们把他们都向 0(虚拟) 节点连边就好

好了,现在思路很明确了

先Tarjan缩点, 然后按缩完的点,以0号节点为边建树

然后直接跑树形DP

代码

#include<bits/stdc++.h>
using namespace std;
#define re register
#define ll long long
#define in inline
#define get getchar()
in int read()
{
    int t=0; char ch=get;
    while(ch<'0' || ch>'9') ch=get;
    while(ch<='9' && ch>='0') t=t*10+ch-'0',ch=get;
    return t;
}
const int _=501;
int n,m,w[_],h[_],d[_],v[_],dfn[_],tot,cnt,color,low[_],top,st[_],vis[_],co[_],ww[_],vv[_],dp[_][_];
struct edge{
    int to,ne;
}e[_*_];
in void add(int x,int y)
{ e[++tot].to=y,e[tot].ne=h[x],h[x]=tot; }

in void Tarjan(int x)
{
    low[x]=dfn[x]=++cnt;
    st[++top]=x,vis[x]=1;
    for(re int i=h[x];i;i=e[i].ne)
    {
        int y=e[i].to;
        if(!dfn[y])
        {
            Tarjan(y);
            low[x]=min(low[x],low[y]);
        }
        else if(vis[y])
            low[x]=min(low[x],dfn[y]);
    }
    if(low[x]==dfn[x])
    {
        ++color;
        while(st[top+1]!=x)
        {
            co[st[top]]=color;
            ww[color]+=w[st[top]];
            vv[color]+=v[st[top]];
            vis[st[top--]]=0;
        }
    }
}

in void dfs(int x)
{
    for(re int i=ww[x];i<=m;i++)
        dp[x][i]=vv[x];
    for(re int i=h[x];;i=e[i].ne)
    {
        if(i==0x3f3f3f3f)break;
        int y=e[i].to;
        dfs(y);
        for(re int j=m-ww[x];j>=0;j--)
            for(re int k=0;k<=j;k++)
                dp[x][j+ww[x]]=max(dp[x][j+ww[x]],dp[y][k]+dp[x][j+ww[x]-k]);
    }
}

int main()
{
    n=read(),m=read();
    for(re int i=1;i<=n;i++)
        w[i]=read();
    for(re int i=1;i<=n;i++)
        v[i]=read();
    for(re int i=1;i<=n;i++)
    {
        d[i]=read();
        if(d[i])
            add(d[i],i);
    }
    for(re int i=1;i<=n;i++)
        if(!dfn[i]) Tarjan(i);
    memset(h,0x3f,sizeof(h));
    memset(vis,0,sizeof(vis));
    memset(e,0,sizeof(e));
    tot=0;
    for(re int i=1;i<=n;i++)
        if(co[d[i]]!=co[i])
        {
            add(co[d[i]],co[i]);
            vis[co[i]]++;
        }
    for(re int i=1;i<=color;i++)
        if(!vis[i])
            add(0,i);
    dfs(0);
    cout<<dp[0][m]<<endl;
    return 0;
}
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!