#include <bits/stdc++.h> using namespace std; typedef long long int LL; const LL MAX_N = 3000005; int n, k, q; LL a[MAX_N], prefSums[MAX_N]; unordered_map<LL, LL> cache; vector<LL> queries; unordered_map<int, int> maxRights; bool DEBUG = false; LL hashPair(const LL left, const LL right) { return left*MAX_N + right; } LL solve(int left, int right) { auto existingIt = cache.find(hashPair(left, right)); if(existingIt != cache.cend()) { return existingIt->second; } vector<LL> dp(n+1); LL maxOverall = 0; LL maxPrev = 0; for(int i = left+k-1; i <= right; ++i) { if(i-k >= left) { maxPrev = max(maxPrev, dp[i-k]); } dp[i] = prefSums[i]-prefSums[i-k]; if(maxPrev > 0) { dp[i] += maxPrev; } if(dp[i] > maxOverall) { maxOverall = dp[i]; } cache[hashPair(left, i)] = maxOverall; //cout << "i: " << i << ", dp[i]: " << dp[i] << endl; } return maxOverall; } int main() { ios_base::sync_with_stdio(0); cin >> n >> k >> q; for(int i = 1; i <= n; ++i) { cin >> a[i]; prefSums[i] = prefSums[i-1] + a[i]; } int l, r; for(int i = 0; i < q; ++i) { cin >> l >> r; queries.push_back(hashPair(l, r)); maxRights[l] = max(maxRights[l], r); } for(const auto& query : maxRights) { solve(query.first, query.second); } for(const auto& query : queries) { cout << cache[query] << endl; } 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 | #include <bits/stdc++.h> using namespace std; typedef long long int LL; const LL MAX_N = 3000005; int n, k, q; LL a[MAX_N], prefSums[MAX_N]; unordered_map<LL, LL> cache; vector<LL> queries; unordered_map<int, int> maxRights; bool DEBUG = false; LL hashPair(const LL left, const LL right) { return left*MAX_N + right; } LL solve(int left, int right) { auto existingIt = cache.find(hashPair(left, right)); if(existingIt != cache.cend()) { return existingIt->second; } vector<LL> dp(n+1); LL maxOverall = 0; LL maxPrev = 0; for(int i = left+k-1; i <= right; ++i) { if(i-k >= left) { maxPrev = max(maxPrev, dp[i-k]); } dp[i] = prefSums[i]-prefSums[i-k]; if(maxPrev > 0) { dp[i] += maxPrev; } if(dp[i] > maxOverall) { maxOverall = dp[i]; } cache[hashPair(left, i)] = maxOverall; //cout << "i: " << i << ", dp[i]: " << dp[i] << endl; } return maxOverall; } int main() { ios_base::sync_with_stdio(0); cin >> n >> k >> q; for(int i = 1; i <= n; ++i) { cin >> a[i]; prefSums[i] = prefSums[i-1] + a[i]; } int l, r; for(int i = 0; i < q; ++i) { cin >> l >> r; queries.push_back(hashPair(l, r)); maxRights[l] = max(maxRights[l], r); } for(const auto& query : maxRights) { solve(query.first, query.second); } for(const auto& query : queries) { cout << cache[query] << endl; } return 0; } |