POJ--1770--Special Experiment--树形DP

互相排斥的点，连一条边，然后就是很正常的树形DP了，，

因为看了discuss里面的说的可能有环的情况，就是不知道数据里面有没有换，所以就接受建议直接用邻接矩阵存了。

#include<iostream>
#include<cstring>
#include<cstdio>
#define maxn 210
#define dbug
#define M(a) memset((a),0,sizeof((a)))
using namespace std;
template<typename T>
T oabs(T a){return a<0?-a:a;}
bool mp[maxn][maxn];
int val[maxn][2];	//val[i][1] indicate include node i
bool vis[maxn];
int n,m;
int a[maxn];
bool  photon[1000000+10];
bool init()
{
	int b;
	M(photon);M(mp);
	photon[0]=1;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=m;i++)
	{
		scanf("%d",&b);
		photon[b]=1;
	}
	return n+m;
}
void DP_process(int u)
{
	vis[u]=-1;
	for(int i=1;i<=n;i++)
	{
		if(u==i)continue;
		if(mp[u][i])
		{
			if(!vis[i])
			{
				DP_process(i);
				val[u][1]+=val[i][0];
				val[u][0]+=max(val[i][1],val[i][0]);
			}
		}
	}
	vis[u]=1;
}
void solve()
{
	M(val);
	for(int i=1;i<=n;i++)
	{
		val[i][1]=a[i];
		for(int j=1;j<=n;j++)
		{
			if(i==j)continue;
			if(photon[oabs(a[i]-a[j])])
			{
				mp[i][j]=mp[j][i]=1;
			}
		}
	}
	M(vis);
	int ans=0;
	for(int i=1;i<=n;i++)
	{
		if(!vis[i])
		{
			DP_process(i);
			ans+=max(val[i][1],val[i][0]);
		}
	}
	printf("%d\n",ans);
}

int main()
{
	while(init())
	solve();
	return 0;
}