[动态规划-4] 合并数

题目:求正数数组内和为指定数字的合并总数

比如num = [5, 5, 10, 2, 3],给定的合并值为 15 :
有4种 : {5 + 10, 5 + 10, 5 + 5 + 2 + 3, 10 + 2 + 3}

分析:

这实际上是网易有道笔试题之一,我觉得我笔试通过主要就是依靠这个题目,因为其他的做的比较一般。

这道题使用动态规划思想,大家看如下的状态转移方程:

dp[n][m]=dp[n-1][m]+dp[n-1][m-num[n-1]];//这里的num[n-1]是第n个数字

dp[n][m]表示前n个元素组成和为m的情况数。初始化dp[0][0]=1,其他为0。写出状态转移方程,大家也就明白了,为何要求全是正数了吧,直白一些,数组的索引,怎么可能为负呢?在计算的过程中,将和的情况保存下来,用空间换时间,整个算法的时间复杂度为O(n*m),不再是指数级。

代码:

#include<iostream>
#include<vector>

using namespace std;

int merge(int num[], int n, int sum)
{
    if(n < 1 || !num || sum<0)
       return 0;
    vector<int> dp[n+1];
    for(int i =0;i<= n;i++)
        dp[i].resize(sum+1);
    
    for(int i = 1;i<=sum;i++)
        dp[0][i] = 0;//用前0个数表示任何非0的数字都有0种方法 
    for(int i = 0;i<=n;i++)
        dp[i][0] = 1;//用数组中的任何数表示0都只有一种方法,那就是什么都不取 
    for(int i = 1;i<=n;i++)
        for(int j = 1;j<=sum;j++)
        {
          int tmp1 = dp[i-1][j];
          int tmp2 = 0;
          if(j - num[i-1]>= 0) 
              tmp2 = dp[i-1][j-num[i-1]];
          dp[i][j] = tmp1 + tmp2;             
        }
    
    //打印出数组来看看更直观,可以省略 
    for(int i = 0;i<=n;i++)
    {
      for(int j = 0;j<=sum;j++)
        printf("%d ",dp[i][j]);
      printf("\n");
    }
    return dp[n][sum];
}



int main()
{
 int num[] = {5,5,10,2,3};

 printf("%d\n",merge(num,5,15));
 system("pause");
 return 0;
}

运行结果如下:



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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值