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<vector>
#warning we made it
using namespace std;
using ll = long long;
using pii = pair<int,long long>;
constexpr int NMAX = 3e5 + 1;
const ll oo = 1e16;
int a[NMAX];
inline pii best(pii a,pii b)
{
if(a.second != b.second) return (a.second > b.second ? a : b);
return (a.first < b.first ? a : b);
}
pii solve(int n,int l)
{
pii dp[2][n + 1]; dp[0][0] = {0,0},dp[1][0] = {0,-oo};
for(int i = 0 ; i < n ; i++)
{
dp[0][i + 1] = best(dp[0][i],dp[1][i]);
dp[1][i + 1] = best({dp[0][i].first + 1,dp[0][i].second + a[i + 1] - l},{dp[1][i].first,dp[1][i].second + a[i + 1]});
}
return best(dp[0][n],dp[1][n]);
}
int main()
{
int n,k; cin >> n >> k;
for(int i = 1; i <= n ; i++) cin >> a[i];
int st = 0 , dr = 1 << 28 , lambda(0);
while(st <= dr)
{
int mid = st + (dr - st) / 2; pii now = solve(n,mid);
if(now.first <= k) dr = mid - 1,lambda = mid;
else st = mid + 1;
}
pii ans = solve(n,lambda);
ll out = ans.second + 1LL * lambda * ans.first;
cout << out;
//solve(n,1);
}
Compilation message (stderr)
feast.cpp:3:2: warning: #warning we made it [-Wcpp]
3 | #warning we made it
| ^~~~~~~
# | 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... |