POJ 1770

POJ 1770

题意：给出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;
}