数组中任取几个数字和为sum的方法数

版权声明:转载请注明出处即可 https://blog.csdn.net/qq_35170212/article/details/81479461

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

手撕dfs失败:

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

void helpf(vector<int>&vr,int pos,int sum,int m,int& res){
    if(sum==m){
        res++;
        return;
    }
    else if(sum>m){
        return;
    }else{
        if(pos<vr.size()){
            sum+=vr[pos];
            helpf(vr,pos+1,sum,m,res);
            sum-=vr[pos];
            helpf(vr,pos+1,sum,m,res);
        }
    }
}

改进为DP:

int main(){
    int n=0;
    int m=0;
    while(cin>>n>>m){
        vector<int> vr(n);
        for(int i=0;i<n;++i){
            cin>>vr[i];
        }
        sort(vr.begin(),vr.end(),greater<int>());
        //      int res=0;
        //      helpf(vr,0,0,m,res);//递归超时,通过率50%
        //      cout<<res<<endl;
        vector<vector<long long int>>dp(n,vector<long long int>(m+1,0));
        for(int i=0;i<n;++i){
            dp[i][0]=1;
        }
        for(int i=1;i<=m;i++){
            if(vr[0]>m)//过滤
                break;
            if(vr[0]==i)
                dp[0][i]=1;
            else
                dp[0][i]=0;
        }
        for(int i=1;i<n;++i){
            if(vr[i]>m)  //过滤
                continue;
            for(int j=1;j<=m;++j){
                if(j-vr[i]>=0)
                    dp[i][j]=dp[i-1][j]+dp[i-1][j-vr[i]];
                else
                    dp[i][j]=dp[i-1][j];
            }
        }
        cout<<dp[n-1][m]<<endl;
    }
    return 0;
}

题目来源:
https://www.nowcoder.com/practice/7f24eb7266ce4b0792ce8721d6259800?tpId=85&tqId=29863&rp=2&ru=/ta/2017test&qru=/ta/2017test/question-ranking

猜你喜欢

转载自blog.csdn.net/qq_35170212/article/details/81479461
今日推荐