sincerit 数字和为sum的方法数(01背包问题)

题目描述
给定一个有n个正整数的数组A和一个整数sum,求选择数组A中部分数字和为sum的方案数。
当两种选取方案有一个数字的下标不一样,我们就认为是不同的组成方案。
输入描述:
输入为两行:
第一行为两个正整数n(1 ≤ n ≤ 1000),sum(1 ≤ sum ≤ 1000)
第二行为n个正整数Ai,以空格隔开。
输出描述:
输出所求的方案数
示例1
输入
复制
5 15 5 5 10 2 3
输出
复制
4

好久没做背包问题了赶紧看一遍优化原理:
https://blog.csdn.net/witnessai1/article/details/52702754

01背包问题优化后的枚举是要从大到小的,因为我们使用的是一维数组
在求dp[v]得时候要使用到以前得数据 即dp[i][v] = max(dp[i-1][v], dp[i-1][v-weight[i]]+cost[i]);
当要求前i个物体背包容量为v时的最大价值,就要用到dp[i-1][v], dp[i-1][v-weight[i]]+cost[i]
也就是在一维数组时求dp[v]要用到dp[v]和dp[v-weight[i]]这两个是i-1层的数据
只能从大到小枚举的原因是:防止dp[v]覆盖i-1层的数据也就是dp[i-1][v]的数据
假设从小到大,那么求dp[v]时, dp[v] = max(dp[v], dp[v-weight[i]]+cost[i])
显然现在求得的dp[v]已经是dp[i][v]了, 可是当后面要求的数据要使用dp[i-1][v]的时候(因为有v-weight[i]这个值比v要小),就不能再用dp[v]因为dp[i-1][v]的值被dp[i][v]覆盖了
因此只能从大到小枚举。

#include <stdio.h>
#include <cstring>
long long dp[1005], A[1005];
// 01背包问题 
// dp[i]表示以和为i的方案数
// 对于A[i] dp[j] = dp[j] + dp[j-A[i]];
// 初始值dp[0] = 1 A[i]=5刚好能凑起的时候 dp[5-5] = 1; 刚好一个5能得到和为5的方案数  
int main() {
  int n, sum;
  scanf("%d %d", &n, &sum);
  for (int i = 0; i < n; i++) scanf("%lld", &A[i]);
  memset(dp, 0, sizeof(dp));
  dp[0] = 1;
  for (int i = 0; i < n; i++) {
    for (int j = sum; j >= A[i]; j--) {
      dp[j] = dp[j] + dp[j - A[i]];
    }
  }
  printf("%lld\n", dp[sum]);
  return 0;
}


未优化的代码:
#include <stdio.h>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
/*
dp[i][j] 表示前i个数其中几个数加起来为j的种类数 
转移方程
dp[i][j] =  dp[i-1][j-num[i]] + dp[i-1][j]
初始值
dp[i][0] = 1; // 表示加起来和为0的种类数(就是都不取) 
dp[0][j] = 0; 
*/
int dp[1000][1000];
int num[1020];
int sum, n, cnt;
int main() {
  scanf("%d %d", &n, &sum);
  for (int i = 1; i <= n; i++) scanf("%d", &num[i]);
  for (int i = 0; i <= n; i++) dp[i][0] = 1; // dp[0][0]也要初始化为1 
  for (int j = 0; j <= sum; j++) dp[0][j] = 0; 
  for (int i = 1; i <= n; i++) {
    for (int j = num[i]; j <= sum; j++) {
        dp[i][j] =  dp[i-1][j-num[i]] + dp[i-1][j];
    }
  }
  printf("%d\n", dp[n][sum]);
}


  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值