#include <iostream>
constexpr long long INF = 1e18;
constexpr int MAX_N = 3e5 + 5;
int N, K, a[MAX_N];
std::pair<long long, int> dp[MAX_N][2];
std::pair<long long, int> check(long long lambda) {
dp[1][0] = { 0, 0 };
dp[1][1] = { a[1] - lambda, 1 };
for (int i = 2; i <= N; ++i) {
dp[i][0] = std::max(dp[i - 1][0], dp[i - 1][1]);
dp[i][1] = std::max(
std::make_pair(dp[i - 1][0].first + a[i] - lambda, dp[i - 1][0].second + 1),
std::make_pair(dp[i - 1][1].first + a[i], dp[i - 1][1].second)
);
}
return std::max(dp[N][0], dp[N][1]);
}
int32_t main() {
std::ios_base::sync_with_stdio(false);
std::cin.tie(nullptr);
std::cin >> N >> K;
for (int i = 1; i <= N; ++i)
std::cin >> a[i];
long long l = 0, r = INF, ans = 0;
while (l <= r) {
long long mid = (l + r) >> 1;
std::pair<long long, int> res = check(mid);
if (res.second >= K) {
ans = res.first + mid * K;
l = mid + 1;
} else r = mid - 1;
}
std::cout << ans;
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
64 ms |
10832 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
47 ms |
10832 KB |
Output is correct |
2 |
Correct |
48 ms |
10832 KB |
Output is correct |
3 |
Correct |
45 ms |
10832 KB |
Output is correct |
4 |
Incorrect |
45 ms |
10832 KB |
Output isn't correct |
5 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
82 ms |
10832 KB |
Output is correct |
2 |
Correct |
84 ms |
10968 KB |
Output is correct |
3 |
Correct |
84 ms |
10824 KB |
Output is correct |
4 |
Correct |
83 ms |
11000 KB |
Output is correct |
5 |
Correct |
86 ms |
10832 KB |
Output is correct |
6 |
Correct |
81 ms |
10832 KB |
Output is correct |
7 |
Correct |
81 ms |
10952 KB |
Output is correct |
8 |
Correct |
83 ms |
10956 KB |
Output is correct |
9 |
Correct |
82 ms |
11036 KB |
Output is correct |
10 |
Correct |
88 ms |
10960 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
2552 KB |
Output is correct |
2 |
Incorrect |
1 ms |
2384 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
2552 KB |
Output is correct |
2 |
Incorrect |
1 ms |
2384 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
1 ms |
2552 KB |
Output is correct |
2 |
Incorrect |
1 ms |
2384 KB |
Output isn't correct |
3 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
64 ms |
10832 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |