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;
}


 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值