互相排斥的点,连一条边,然后就是很正常的树形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;
}