笔试刷题-滴滴2018-06-16

2018-06-16  本文已影响0人  Dodo159753

题目如下:


/**
给定一个有n个正整数的数组A和一个整数sum,求选择数组A中部分数字和为sum的方案数。
当两种选取方案有一个数字的下标不一样,我们就认为是不同的组成方案。
输入描述:
输入为两行:
 第一行为两个正整数n(1 ≤ n ≤ 1000),sum(1 ≤ sum ≤ 1000)
 第二行为n个正整数A[i](32位整数),以空格隔开。


输出描述:
输出所求的方案数

输入例子1:
5 15 5 5 10 2 3

输出例子1:
4

*/ 

思路如下:

方案一:搜索+记忆化(N太大)

方案二:DP + 滚动数组减少空间

代码如下:


#include<stdio.h>
#include<iostream>

#define MAX_N 1005
#define MAX_SUM 1005

typedef long long LL;

using namespace std;

int values[MAX_N];
LL dp[MAX_N][MAX_SUM];

int main(){
    int N, Sum;
    scanf("%d%d", &N, &Sum);
    for(int i=1; i<=N; i++)
        scanf("%d", values+i);
    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>=values[i])
                dp[i][j]+=dp[i-1][j-values[i]];
        }
    }
    printf("%lld", dp[N][Sum]);
    return 0;
}


上一篇 下一篇

猜你喜欢

热点阅读