题意:给出N个点,再给出M个数,要求找出一个点集,点的权值总和要最大,而且点集里面任何2个点之差不等于M里面的任何数。
看起来似乎是最大点权独立集,一点都不会,后来去翻人家的答案,说是题目暗示了其实是一棵树,不存在回路,所以用树状DP可以搞定。
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<stdio.h>
#include<string>
#include<cstring>
#include<map>
#include<vector>
#include<set>
using namespace std;
#define LL long long
#define inf 0x7ffffff
#define N 205
int dp[205][3];//1代表不取该点最大值,2代表取该点最大值
int s[N][N];
int a[N],b[N];
int mark[N];int m,n;
map<int,int>atom;
vector<int>v;
void deal()
{
memset(dp, 0 ,sizeof(dp));
memset(s, 0 ,sizeof(s));
memset(mark, 0 ,sizeof(mark));
atom.clear();
v.clear();
}
void show()
{
for(int i =1;i <= n;i++)
{
for(int j = 1;j <= n;j++)
{
cout<<s[i][j]<<' ';
}
cout<<endl;
}
}
int main()
{
int i,j,k;
while( cin>> n>> m && n + m)
{
deal();
for(i = 1;i <= n;i++)
{
cin>>a[i];
atom[a[i]] = i;
}
for(j = 1;j <= m;j++)
{
cin>>b[j];
}
for(i = 1;i <= n;i++)
{
for(j = 1;j <= m;j++)
{
// cout<<"data "<<endl;
// cout<<a[i]<<' '<<b[j]<<endl;
if(atom.find(a[i] + b[j]) != atom.end())
{
// cout<<i<<' '<<atom[a[i] + b[j]]<<endl;
s[i][atom[a[i] + b[j]]] = 1;
}
if(atom.find(a[i] - b[j]) != atom.end())
{
// cout<<i<<' '<<atom[a[i] - b[j]]<<endl;
s[i][atom[a[i] - b[j]]] = 1;
}
}
}
// show();
int sum = 0;
for(i = 1;i <= n;i++)
{
if(mark[i] == 0)
{
mark[i] = 1;
v.clear();
v.push_back(i);
for(j = 0;j < v.size();j++)
{
for(k = 1;k <= n;k++)
{
if(s[v[j]][k] == 1 && mark[k] == 0)
{
mark[k] = 1;
v.push_back(k);
}
}
}
for(j = v.size() - 1;j >= 0;j--)
{
dp[v[j]][0] = 1;
dp[v[j]][2] += a[v[j]];
for(k = 1;k <= n;k++)
{
if(dp[k][0] == 1 && s[v[j]][k] == 1)
{
dp[v[j]][2] += dp[k][1];
dp[v[j]][1] += max(dp[k][1],dp[k][2]);
}
}
}
sum += max(dp[v[0]][1],dp[v[0]][2]);
}
}
cout<<sum<<endl;
}
return 0;
}