jzoj1154【GDOI2003】购物（树形dp）

1154. 【GDOI2003】购物

Description
　　GDOI商场推出优惠活动，以超低价出售若干种商品。但是，商场为避免过分亏本，规定某些商品不能同时购买，而且每种超低价商品只能买一件。身为顾客的你想获得最大的实惠，也就是争取节省最多的钱。经过仔细研究，发现商场出售的超低价商品中，不存在以下情况： 　　n(n>=3)种商品C1,C2,…..,Cn,其中Ci,Ci+1是不能同时购买的(i=1,2…,n-1)并且C1, Cn也不能同时购买。 　　编程计算可以节省的最大金额数。
Input
　　第一行两个整数K,M(1<=K<=1000).其中K表示超低价商品数。K种商品的编号依次为1,2,…,K。M表示不能同时购买的商品对数.接下来K行，第i行有一个整数Xi表示购买编号为i的商品可以节省的金额(1<=Xi<=100).再接下来M行，每行两个数A ,B,表示A和B不能同时购买，1<=A<=K,1<=B<=K,A<>B
Output
　　仅一个整数，表示能节省的最大金额数。

分析：把每对不能同时购买的商品连边，由于不存在环，所以会组成很多棵树，
那么就可以用树形dp，f[i][0]表示以第i个节点为根的子树不选i时的最大值，f[i][1]表示以第i个节点为根的子树选i时的最大值，显然可得方程
f[i][0]=Σmax(f[son][0],f[son][1])
f[i][1]=c[i]+Σf[son][0]

代码

#include <cstdio>
#define maxn 20000
using namespace std;

int f[maxn][2],ls[maxn],next[maxn],x[maxn],y[maxn],c[maxn];
int n,m,o=0,ans;
bool v[maxn];

void add(int i,int j)
{
    o++;
    x[o]=i;
    y[o]=j;
    next[o]=ls[i];
    ls[i]=o;
}

int dfs(int i)
{
    v[i]=true;
    for (int j=ls[i];j;j=next[j])
        if (!v[y[j]])
        {
            dfs(y[j]);
            f[i][0]+=f[y[j]][1]>f[y[j]][0]?f[y[j]][1]:f[y[j]][0]; 
            f[i][1]+=f[y[j]][0];
        }
    f[i][1]+=c[i];
    return 0;
}

int main()
{
    scanf("%d%d",&n,&m);
    ans=0;
    for (int i=1;i<=n;i++)
        scanf("%d",&c[i]);
    for (int i=1;i<=m;i++)
    {
        int p,q;
        scanf("%d%d",&p,&q);
        add(p,q);
        add(q,p);
    }
    for (int i=1;i<=n;i++)
        if (!v[i]) 
        {
            dfs(i);
            ans+=f[i][0]>f[i][1]?f[i][0]:f[i][1];
        }
    printf("%d",ans);
    return 0;
}