牛客网-2017年校招真题训练--数字和为sum的方法数

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

【input】5 15 5 5 10 2 3
【output】4

典型的dp问题。

代码中dp(i,j) 代表使用数组A中前 i 个数字组成和为 j 的方案数。

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<memory.h>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<vector>
#include<queue>
#define mem(array)  memset((array),0,sizeof((array)))
#define Qsort(array,len,cmp) qsort(array,len,sizeof(array[0]),cmp)
#define inf 0x7fffffff
#define MAXN 10+1000
using namespace std;

int a[MAXN];
long long dp[MAXN][MAXN];

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",tdout);

    int n,sum;
    while(cin>>n>>sum){
        for(int i = 1; i <= n; ++i)
            scanf("%d",&a[i]);

        mem(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){
                if(a[i] <= j)
                    dp[i][j] = dp[i-1][j] + dp[i-1][j-a[i]];
                else
                    dp[i][j] = dp[i-1][j];
            }
        }
        cout<<dp[n][sum]<<endl;
    }
    return 0;
}

题目来源于  牛客网