This submission is migrated from previous version of oj.uz, which used different machine for grading. This submission may have different result if resubmitted.
#include <iostream>
#include <fstream>
#include <vector>
#include <queue>
#include <utility>
#include <algorithm>
#include <cstring>
int x, k;
int arr[300005];
std::pair<int64_t,int> dp[300005][2];
std::pair<int64_t,int> sol(int64_t lambda) {
dp[0][0] = {0,0};
dp[0][1] = {arr[0]-lambda,1};
for (int i = 1; i < x; i++) {
dp[i][0] = std::max(dp[i-1][0],dp[i-1][1]);
dp[i][1] = std::max(std::pair<int64_t,int>(dp[i-1][0].first+arr[i]-lambda,dp[i-1][0].second+1),std::pair<int64_t,int>(dp[i-1][1].first+arr[i],dp[i-1][1].second));
}
return std::max(dp[x-1][0],dp[x-1][1]);
}
int main() {
std::ios_base::sync_with_stdio(false);
std::cin.tie(NULL);
std::cout.tie(NULL);
std::cin >> x >> k;
for (int i = 0; i < x; i++) {
std::cin >> arr[i];
}
const auto bs = [&]() {
int64_t l = 0, r = 1000000000000000000;
int64_t ret = 69420;
while (l<=r) {
int64_t m = (l+r)>>1;
if (sol(m).second>=k) {
ret = m;
l = m+1;
}
else {
r = m-1;
}
}
return ret;
};
int64_t ret = bs();
std::cout << sol(ret).first + ret*k << "\n";
}
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|
Fetching results... |