diff options
| author | Lucas Schwebler <lucas.schwebler@gmail.com> | 2024-09-10 23:06:28 +0200 |
|---|---|---|
| committer | Lucas Schwebler <lucas.schwebler@gmail.com> | 2024-09-10 23:06:28 +0200 |
| commit | facc5da35282ef30e5111cdc04942d118f4ae0c5 (patch) | |
| tree | 7c568dcfbc0da8e4ad6ae3188bb4bc4e657d7dba /content/other/fastSubsetSum.cpp | |
| parent | 36fa19a38cf9a357f04d4ed76f25b1cbf44deedb (diff) | |
add fastSubsetSum
Diffstat (limited to 'content/other/fastSubsetSum.cpp')
| -rw-r--r-- | content/other/fastSubsetSum.cpp | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/content/other/fastSubsetSum.cpp b/content/other/fastSubsetSum.cpp new file mode 100644 index 0000000..84396f6 --- /dev/null +++ b/content/other/fastSubsetSum.cpp @@ -0,0 +1,21 @@ +int fastSubsetSum(vector<int> w, int t){ + int a = 0, b = 0; + while(b < sz(w) && a + w[b] <= t) a += w[b++]; + if(b == sz(w)) return a; + int m = *max_element(all(w)); + vector<int> dp(2*m, -1), old; + dp[m+a-t] = b; + for(int i = b; i < sz(w); i++){ + old = dp; + for(int j = 0; j < m; j++){ + dp[j+w[i]] = max(dp[j+w[i]], old[j]); + } + for(int j = 2*m-1; j > m; j--){ + for(int k = max(old[j], 0); k < dp[j]; k++){ + dp[j-w[k]] = max(dp[j-w[k]], k); + } + } + } + for(a = t; dp[m+a-t] < 0; a--); + return a; +}
\ No newline at end of file |
