POJ 1770 树形DP

题意：

容器中有 n 种原子，现提供 m 种光子，这 n 种原子中能量之差等于提供的 m 种光子之一所具有的能量，就认为这两种原子是处在容器在是危险的，现在要求取走一些原子，使得容器中的原子处在一块是安全的，同时要求剩下的原子的能量和最大。

 

思路：

很明显的树形dp，每两个处于会产生危险的原子连一条无向边，注意建图完了可能是一个森林！

PS：不会有环么？我AC了以后才发现这个问题。。。网上都没提这事，不知道我没完全理解题？

 

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <cmath>
#include <map>

#define N 10000
#define M 100000

using namespace std;

map<int,int> mp;

int head[N],to[M],next[M],a[N],b[N],cnt,num,n,m,tc[N],dp[N][2];
bool vis[N];

inline void add(int u,int v)
{
    to[cnt]=v; next[cnt]=head[u]; head[u]=cnt++;
}

void read()
{
    mp.clear();
    memset(head,-1,sizeof head);cnt=0;
    num=0;
    for(int i=1;i<=n;i++) scanf("%d",&a[i]);
    for(int i=1;i<=m;i++) scanf("%d",&b[i]);
    for(int i=1;i<=n;i++)
        for(int j=i+1;j<=n;j++)
            for(int k=1;k<=m;k++)
                if(abs(a[i]-a[j])==b[k])
                {
                    if(mp[a[i]]==0) {mp[a[i]]=++num;tc[num]=a[i];}
                    if(mp[a[j]]==0) {mp[a[j]]=++num;tc[num]=a[j];}
                    int ta=mp[a[i]],tb=mp[a[j]]; 
                    add(ta,tb),add(tb,ta);
                }
}

void find(int u,int fa)
{
    vis[u]=true;
    for(int i=head[u];~i;i=next[i])
        if(to[i]!=fa) find(to[i],u);
        
    dp[u][1]=tc[u];dp[u][0]=0;
    for(int i=head[u];~i;i=next[i])
        if(fa!=to[i])
        {
            dp[u][0]+=max(dp[to[i]][1],dp[to[i]][0]);
            dp[u][1]+=dp[to[i]][0];
        }
}

void go()
{
    memset(dp,0x8f,sizeof dp);
    memset(vis,0,sizeof vis);
    int ans=0;
    for(int i=1;i<=num;i++)
        if(!vis[i])
        {
            find(i,-1);
            ans+=max(dp[i][0],dp[i][1]);
        }
    printf("%d\n",ans);
}

int main()
{
    while(scanf("%d%d",&n,&m),n||m)
    {
        read();
        go();
    }
    return 0;
}