hdu 5616 Jam's balance(简单dp)

最新推荐文章于 2020-04-23 19:34:38 发布

原创最新推荐文章于 2020-04-23 19:34:38 发布 · 465 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#dp

dp 专栏收录该内容

106 篇文章

订阅专栏

本文详细介绍了使用动态规划解决HDU 5616问题的方法，通过构建dp数组来记录每一步可能达到的状态，最终判断目标值是否可以通过加减运算得到。

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5616

解题思路：

动态规划思想：dp[i][j]表示前i个数通过加减法是否可以等于j，如果为1，则表示可以，0表示不行。

定义好状态，那么这个模型跟简单的01背包模型一样。。。水过。。。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

int n,m,q,sum,w[25];
int dp[25][2005];

void init()
{
	memset(dp,0,sizeof(dp));
	for(int i = 0; i <= n; i++) dp[i][0] = 1;
	for(int i = 1; i <= n; i++)
		for(int j = 1; j <= sum; j++)
		{
			dp[i][j] = dp[i-1][j];
			if(j - w[i] >= 0 && dp[i-1][j-w[i]] == 1)
				dp[i][j] = 1;
			if(j + w[i] <= sum && dp[i-1][j+w[i]] == 1)
				dp[i][j] = 1;
		}
}

int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d",&n);
		sum = 0;
		for(int i = 1; i <= n; i++)
		{
			scanf("%d",&w[i]);
			sum += w[i];
		}
		init();
		scanf("%d",&q);
		while(q--)
		{
			scanf("%d",&m);
			bool flag = false;
			for(int i = 1; i <= n; i++)
				if(dp[i][m] == 1)
				{
					flag = true;
					break;
				}
			if(flag) printf("YES\n");
			else printf("NO\n");
		}
	}
	return 0;
}