多重部分和问题

来源：互联网发布：json数据转换成数组编辑：程序博客网时间：2024/05/19 14:53

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

输入：

3 17

3 5 8

3 2 2

输出：

Yes

#include <cstdio>#include <cstring>#include <algorithm>using namespace std;int n, k, a[1000], m[1000];int dp[1000];void solve(){    memset(dp, -1, sizeof(dp));    dp[0] = 0;    for (int i = 0; i < n; i++){        for (int j = 0; j <= k; j++){            if (dp[j] >= 0){                dp[j] = m[i];            }            else if (j < a[i] || dp[j - a[i]] <= 0){                dp[j] = -1;            }            else{                dp[j] = dp[j - a[i]] - 1;            }        }    }    if (dp[k] >= 0)        printf("Yes\n");    else        printf("No\n");}int main(){    while (scanf("%d%d", &n, &k) != EOF){        for (int i = 0; i < n; i++){            scanf("%d", &a[i]);        }        for (int i = 0; i < n; i++){            scanf("%d", &m[i]);        }        solve();    }    return 0;}

0 0