UVa 10616 Divisible Group Sums (DFS&DP)

最新推荐文章于 2017-11-11 20:42:11 发布

原创最新推荐文章于 2017-11-11 20:42:11 发布 · 1k 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #UVa #acm

UVa 同时被 3 个专栏收录

407 篇文章

订阅专栏

acm之路--好题/陷阱

154 篇文章

订阅专栏

acm之路--DP

76 篇文章

订阅专栏

本文介绍了一种使用深度优先搜索（DFS）结合记忆化搜索的方法来解决 UVA 10616 Divisible Group Sums 问题。该问题要求从一组整数中找出所有可能的子集，这些子集的元素数量等于给定的 m，并且子集的元素之和能被另一个给定的数 d 整除。通过递归调用实现枚举所有可能的组合。

10616 - Divisible Group Sums

Time limit: 3.000 seconds

http://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&category=115&page=show_problem&problem=1557

思路：用DFS+记忆化搜索枚举组合，注意数字有<0的。

return dp[i][cnt][sum] = f(i + 1, cnt + 1, ((sum + arr[i]) % d + d) % d) + f(i + 1, cnt, sum);///i<n,cnt<m

完整代码：

/*0.055s*/

#include<cstdio>
#include<cstring>
typedef long long ll;

ll dp[205][15][25];
int d, m, n, arr[205];

ll f(int i, int cnt, ll sum)
{
	if (cnt == m) return dp[i][cnt][sum] = (sum % d ? 0 : 1);///选m个
	if (i == n) return dp[i][cnt][sum] = 0;///总共n个，如果i==n，说明cnt!=m，也就是还没有选完，此时dp=0
	if (dp[i][cnt][sum] >= 0) return dp[i][cnt][sum];
	return dp[i][cnt][sum] = f(i + 1, cnt + 1, ((sum + arr[i]) % d + d) % d) + f(i + 1, cnt, sum);
}

int main()
{
	int cas = 0, q, i, ans;
	while (scanf("%d%d", &n, &q), n)
	{
		for (i = 0; i < n; ++i) scanf("%d", &arr[i]);
		printf("SET %d:\n", ++cas);
		for (i = 1; i <= q; ++i)
		{
			ans = 0;
			memset(dp, -1, sizeof(dp));
			scanf("%d%d", &d, &m);
			printf("QUERY %d: %d\n", i, f(0, 0, 0));
		}
	}
	return 0;
}