数字和为sum的方法数-01背包计数问题
生活随笔
收集整理的這篇文章主要介紹了
数字和为sum的方法数-01背包计数问题
小編覺得挺不錯的,現在分享給大家,幫大家做個參考.
給定一個有n個正整數的數組A和一個整數sum,求選擇數組A中部分數字和為sum的方案數。
當兩種選取方案有一個數字的下標不一樣,我們就認為是不同的組成方案。
輸入描述:
輸入為兩行:
第一行為兩個正整數n(1 ≤ n ≤ 1000),sum(1 ≤ sum ≤ 1000)
第二行為n個正整數Ai,以空格隔開。
輸出描述:
輸出所求的方案數
示例1
輸入
5 15 5 5 10 2 3
輸出
4
#include <iostream> using namespace std; const int N = 1010; int dp[N][N]; int a[N];int main() {int n, k;cin >> n >> k;for (int i = 1; i <= n; i++)cin >> a[i];dp[0][0] = 1;for (int i = 1; i <= n; i++)for (int j = 0; j <= k; j++) {dp[i][j] = dp[i - 1][j];if (j >= a[i])dp[i][j] += dp[i - 1][j - a[i]];}cout << dp[n][k] << endl;return 0; }空間優化:
#include <iostream> using namespace std; const int N = 1010; int dp[N]; int a[N];int main() {int n, k;cin >> n >> k;dp[0] = 1;for (int i = 1; i <= n; i++)cin >> a[i];for (int i = 1; i <= n; i++)for (int j = k; j >= a[i]; j--) {dp[j] += dp[j - a[i]];}cout << dp[k] << endl;return 0; } 創作挑戰賽新人創作獎勵來咯,堅持創作打卡瓜分現金大獎總結
以上是生活随笔為你收集整理的数字和为sum的方法数-01背包计数问题的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: 总是因为图表问题被返修?不看后悔系列的S
- 下一篇: [蓝桥杯][算法提高VIP]数的划分-d