#include<bits/stdc++.h> using namespace std; const int N = 205; const long long inf = 2e18 + 10; long long tbl[N]; map<pair<pair<int, int>, long long>, long long> M; long long powerOf2(long long k) { if (k == 0) return 0; long long result = 1; while(result * 2 <= k) result *= 2; return result; } int maximumNumberOfBits(long long k) { k++; int result = 0; while (k > 1) k /= 2, result++; return result; } long long calc(int l, int p, long long k) { if (p < l || k < p - l) return -inf; auto it = M.find({{l, p}, k}); if (it != M.end()) { return it->second; } if (l == p) { long long result = max(0ll, tbl[p] - tbl[l - 1]) * maximumNumberOfBits(k); M[{{l, p}, k}] = result; return result; } long long mediumValue = powerOf2(k) - 1; long long maxResult = -inf; for (int mid = l - 1; mid <= p; mid++) { maxResult = max(maxResult, calc(l, mid, mediumValue) + calc(mid + 1, p, k - mediumValue - 1) + tbl[p] - tbl[mid]); } M[{{l, p}, k}] = maxResult; return maxResult; } void solve() { int n; long long k; cin >> n >> k; for (int i = 1; i <= n; i++) { cin >> tbl[i]; tbl[i] += tbl[i - 1]; } /*for (int i = 0; i <= k; i++) { cout << i << " " << maximumNumberOfBits(i) << " " << powerOf2(i) << "\n"; }*/ cout << calc(1, n, k); return; } int32_t main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int t = 1; //cin >> t; while(t--) solve(); return 0; }
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 | #include<bits/stdc++.h> using namespace std; const int N = 205; const long long inf = 2e18 + 10; long long tbl[N]; map<pair<pair<int, int>, long long>, long long> M; long long powerOf2(long long k) { if (k == 0) return 0; long long result = 1; while(result * 2 <= k) result *= 2; return result; } int maximumNumberOfBits(long long k) { k++; int result = 0; while (k > 1) k /= 2, result++; return result; } long long calc(int l, int p, long long k) { if (p < l || k < p - l) return -inf; auto it = M.find({{l, p}, k}); if (it != M.end()) { return it->second; } if (l == p) { long long result = max(0ll, tbl[p] - tbl[l - 1]) * maximumNumberOfBits(k); M[{{l, p}, k}] = result; return result; } long long mediumValue = powerOf2(k) - 1; long long maxResult = -inf; for (int mid = l - 1; mid <= p; mid++) { maxResult = max(maxResult, calc(l, mid, mediumValue) + calc(mid + 1, p, k - mediumValue - 1) + tbl[p] - tbl[mid]); } M[{{l, p}, k}] = maxResult; return maxResult; } void solve() { int n; long long k; cin >> n >> k; for (int i = 1; i <= n; i++) { cin >> tbl[i]; tbl[i] += tbl[i - 1]; } /*for (int i = 0; i <= k; i++) { cout << i << " " << maximumNumberOfBits(i) << " " << powerOf2(i) << "\n"; }*/ cout << calc(1, n, k); return; } int32_t main() { ios_base::sync_with_stdio(0); cin.tie(0); cout.tie(0); int t = 1; //cin >> t; while(t--) solve(); return 0; } |