#include <bits/stdc++.h>
#define eb emplace_back
#define mp make_pair
#define F first
#define S second
#define all(x) x.begin(), x.end()
#define svec(x) sort(x.begin(), x.end())
#define press(x) x.erase(unique(x.begin(), x.end()), x.end())
#define lb(x, v) lower_bound(x.begin(), x.end(), v)
#define ub(x, v) upper_bound(x.begin(), x.end(), v)
using namespace std;
typedef long long LL;
typedef pair<int, int> pii;
typedef pair<LL, LL> pll;
typedef pair<int, LL> pil;
typedef pair<LL, int> pli;
const LL llinf=10000000000000;
const LL mod1=1000000007;
const LL mod2=998244353;
const int inf=2000000000;
int n, k;
LL arr[300010], dp[300010][2], cnt[300010][2];
int solve(LL x){
dp[0][0]=-llinf;
dp[0][1]=0;
for(int i=1; i<=n; i++){
if(dp[i-1][0]>dp[i-1][1]-x)dp[i][0]=dp[i-1][0]+arr[i], cnt[i][0]=cnt[i-1][0];
else dp[i][0]=dp[i-1][1]+arr[i]-x, cnt[i][0]=cnt[i-1][1]+1;
tie(dp[i][1], cnt[i][1])=max(tie(dp[i-1][0], cnt[i][0]), tie(dp[i-1][1], cnt[i][1]));
}
if(pll(dp[n][0], -cnt[n][0])>pll(dp[n][1], -cnt[n][1]))return cnt[n][0];
return cnt[n][1];
}
int main(){
scanf("%d %d", &n, &k);
for(int i=1; i<=n; i++){
scanf("%lld", &arr[i]);
arr[i]*=2;
}
LL l=0, r=llinf;
while(l<r){
LL mid=(l+r)/2;
if(solve(mid*2+1)>k)l=mid+1;
else r=mid;
}
solve(l*2);
printf("%lld", max(dp[n][0], dp[n][1])/2+l*k);
}
Compilation message
feast.cpp: In function 'int main()':
feast.cpp:38:10: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%d %d", &n, &k);
~~~~~^~~~~~~~~~~~~~~~~
feast.cpp:40:14: warning: ignoring return value of 'int scanf(const char*, ...)', declared with attribute warn_unused_result [-Wunused-result]
scanf("%lld", &arr[i]);
~~~~~^~~~~~~~~~~~~~~~~
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
117 ms |
11640 KB |
Output is correct |
2 |
Correct |
110 ms |
11896 KB |
Output is correct |
3 |
Correct |
111 ms |
12024 KB |
Output is correct |
4 |
Correct |
116 ms |
11896 KB |
Output is correct |
5 |
Correct |
114 ms |
11896 KB |
Output is correct |
6 |
Correct |
114 ms |
11644 KB |
Output is correct |
7 |
Correct |
109 ms |
11804 KB |
Output is correct |
8 |
Correct |
120 ms |
11896 KB |
Output is correct |
9 |
Correct |
118 ms |
11828 KB |
Output is correct |
10 |
Correct |
119 ms |
11896 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
89 ms |
11768 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
181 ms |
12024 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
5 ms |
384 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
5 ms |
384 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
5 ms |
384 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
117 ms |
11640 KB |
Output is correct |
2 |
Correct |
110 ms |
11896 KB |
Output is correct |
3 |
Correct |
111 ms |
12024 KB |
Output is correct |
4 |
Correct |
116 ms |
11896 KB |
Output is correct |
5 |
Correct |
114 ms |
11896 KB |
Output is correct |
6 |
Correct |
114 ms |
11644 KB |
Output is correct |
7 |
Correct |
109 ms |
11804 KB |
Output is correct |
8 |
Correct |
120 ms |
11896 KB |
Output is correct |
9 |
Correct |
118 ms |
11828 KB |
Output is correct |
10 |
Correct |
119 ms |
11896 KB |
Output is correct |
11 |
Incorrect |
89 ms |
11768 KB |
Output isn't correct |
12 |
Halted |
0 ms |
0 KB |
- |