多重部分和问题

多重部分和问题

问题描述

有n种不同大小的数字ai，每种有mi个，判断是否可以从中选若干个数字使它们的和恰好为k

限制条件：

1<=n<=100;

1<=ai , mi<=100000;

1<=k<=100000;

输入：

第一行 n，k

后n行  ai , mi

3 17

3 3

5 2

8 2

输出：

Yes

类似于完全背包问题，给出DP求解

#include<iostream>
#include<string.h>
using namespace std;
#define N 100
#define M 100000

int dp[N][M];

int main()
{
    int n, k;
    int a[N], m[N];
    int i, j, h;
    cin>>n>>k;
    for(i = 0; i < n; i++)
        cin>>a[i]>>m[i];

    memset(dp, 0, sizeof(dp));
    dp[0][0] = 1;
    for(i = 0; i < n; i++)
    {
        for(j = 0; j <= k; j++)
        {
            for(h = 0; h <= m[i] && h*a[i]<=j; h++)
                dp[i+1][j] += dp[i][j-h*a[i]];
        }
    }

    for(i = 0; i < n; i++)
    {
        for(j = 0; j < k; j++)
        {
            cout<<dp[i][j]<<" ";
        }
        cout<<endl;
    }

    if(dp[n-1][k-1])
        cout<<"Yes"<<endl;
    else
        cout<<"No"<<endl;
    return 0;
}

这个方式同时计算了和为k的数字取法有多少种！

但是效率并不高！