BZOJ 2427 软件安装(HAOI 2010)

好吧，说真的，作为一个，尤其是刚学树形dp的蒟蒻，做这个题目的时候，我的内心是崩溃的。这个题目做了好久，虽说已经想到了思路，结果码完代码之后，总是WA，于是就暂时丢到一边去了。今天有空，自己做了几份（十分水的）数据手测，我果然并没有发现程序错在哪。
后来突然发现，自己特别蠢地写了个儿子指针的tree结构体来保存树，结果儿子又必须得开到n，很容易MLE，于是便只好又重新码了一遍……

题目详细描述就先草率地略过了~

Sample Input

3 10
5 5 6
2 3 4
0 1 1

Sample Output

5

思路

其实，本题的思路还是算比较简单清晰的。首先既然每个软件都有占用空间和相应的价值，那么想必就是一个01背包问题。
其次由于某些软件有可能会互相依赖，就构成了一个环，对于一个环，要想其价值不为0，那么必须要装全套，否则必定是不利的，（XXX电脑安全套装，你值得拥有），那么就可以将这个环看作一个节点。
另外值得一提的是，这个题目的设定是一个软件最多有一个依赖对象，所以也可以针对这一特征进行判环操作，但本文中用的是tarjan。
由此，我们得出做法：判环缩点+树上01背包；

代码

#include <iostream>
#include <cstring>
#include <cstdio>
#include <stack>
using namespace std;
const int N=110,M=510;
struct date{
    int ptr,nxt; 
}edge[M],nedge[M];
int w[N],c[N],nw[N],nc[N],d[N],head[N],nhead[N];
int n,m,temp,p,np,tot,total,low[M],dfn[N],f[N][M];
bool vis[N];
stack<int> s;
template <typename Tp> void read(Tp &x)
{
    x=0;
    char ch=getchar();
    while(ch<'0'||ch>'9') ch=getchar();
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
}
int min(int x,int y){return x<y?x:y;}
int max(int x,int y){return x>y?x:y;}
void add(int from,int to,bool flag)
{
    if(!flag)
    {
        edge[++p].nxt=head[from];
        edge[p].ptr=to;
        head[from]=p;
        return ;
    }
    nedge[++np].nxt=nhead[from];
    nedge[np].ptr=to;
    nhead[from]=np;
}
void tarjan(int x)
{
    int xx;
    s.push(x);
    vis[x]=true;
    dfn[x]=low[x]=(++tot);
    for(int i=head[x];i;i=edge[i].nxt)
    {
        xx=edge[i].ptr;
        if(!dfn[xx])
        {
            tarjan(xx);
            low[x]=min(low[x],low[xx]);
        }
        if(vis[xx])
          low[x]=min(low[x],dfn[xx]);
    }
    if(low[x]==dfn[x])
    {
        total++;
        do{
            temp=s.top();
            s.pop();
            vis[temp]=false;
            d[temp]=total;
            nc[total]+=c[temp];
            nw[total]+=w[temp];
        }while(temp!=x);
    }
}
void build()
{
    for(int i=1;i<=n;i++)
      for(int j=head[i];j;j=edge[j].nxt)
        if(d[i]!=d[edge[j].ptr])
        {
            vis[d[edge[j].ptr]]=true;
            add(d[i],d[edge[j].ptr],true);
        }
    for(int i=1;i<=total;i++)
      if(!vis[i])
      {
        vis[i]=true;
        add(total+1,i,true);
      }
}
void calc(int x)
{
    for(int i=nhead[x];i;i=nedge[i].nxt)
      calc(nedge[i].ptr);
    for(int i=nhead[x];i;i=nedge[i].nxt)
      for(int j=m-nw[x];j>=0;j--)
        for(int r=0;r<=j;r++)
          f[x][j]=max(f[x][j],f[x][r]+f[nedge[i].ptr][j-r]);
    for(int j=m;j>=0;j--)
      f[x][j]=(j>=nw[x]?f[x][j-nw[x]]+nc[x]:0);
}
int main()
{
    //freopen("install.in","r",stdin);
    read(n),read(m);
    for(int i=1;i<=n;i++)
      read(w[i]);
    for(int i=1;i<=n;i++)
      read(c[i]);
    for(int i=1;i<=n;i++)
    {
        read(temp);
        if(temp!=0)
          add(temp,i,false);
    }
    for(int i=1;i<=n;i++)//判环缩点
      if(!dfn[i])
        tarjan(i);
    memset(vis,0,sizeof(vis));
    build();//重新建树
    calc(total+1);
    printf("%d",f[total+1][m]);
    return 0;
}