#include <bits/stdc++.h>
using namespace std;
#define pb push_back
#define endl '\n'
#define ll long long
#define int long long
const long long inf=1e18;
const int MOD=998244353;
const int N=3e5+5;
int n,k,a[N];
pair<int,int> dp[N][2];
int func(int lambda){
dp[0][0]={0,0};//empty
dp[0][1]={-lambda,1};//using it
for(int i=1;i<=n;i++){
dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
int x=dp[i-1][0].first-lambda+a[i],y=dp[i-1][1].first+a[i];
if(x>y || (x==y && dp[i-1][0]<dp[i-1][1])){
dp[i][1].first=x;
dp[i][1].second=dp[i-1][0].second+1;
}
else{
dp[i][1].first=y;
dp[i][1].second=dp[i-1][1].second;
}
}
int x=dp[n][0].first,y=dp[n][1].first;
if(x==y && dp[n][1]<dp[n][0])swap(dp[n][0],dp[n][1]);
return (x>y?dp[n][0].second:dp[n][1].second);
}
signed main() {
ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);
cin>>n>>k;
for(int i=1;i<=n;i++)cin>>a[i];
int l=0,r=1e18;
while(l<r){
int mid=(l+r+1)/2;
int x=func(mid);
if(x>=k)l=mid;
else r=mid-1;
}
func(l);
cout<<max(dp[n][0].first+l*dp[n][0].second,dp[n][1].first+l*dp[n][1].second);
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
79 ms |
11740 KB |
Output is correct |
2 |
Correct |
86 ms |
11872 KB |
Output is correct |
3 |
Correct |
55 ms |
12168 KB |
Output is correct |
4 |
Correct |
52 ms |
12124 KB |
Output is correct |
5 |
Correct |
51 ms |
11864 KB |
Output is correct |
6 |
Correct |
51 ms |
11612 KB |
Output is correct |
7 |
Correct |
51 ms |
11604 KB |
Output is correct |
8 |
Correct |
52 ms |
11952 KB |
Output is correct |
9 |
Correct |
51 ms |
11848 KB |
Output is correct |
10 |
Correct |
62 ms |
11968 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
45 ms |
11864 KB |
Output is correct |
2 |
Correct |
47 ms |
12120 KB |
Output is correct |
3 |
Correct |
45 ms |
11864 KB |
Output is correct |
4 |
Correct |
45 ms |
11864 KB |
Output is correct |
5 |
Correct |
53 ms |
11828 KB |
Output is correct |
6 |
Correct |
50 ms |
11868 KB |
Output is correct |
7 |
Correct |
45 ms |
12124 KB |
Output is correct |
8 |
Correct |
52 ms |
12124 KB |
Output is correct |
9 |
Correct |
52 ms |
11612 KB |
Output is correct |
10 |
Correct |
45 ms |
11864 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
69 ms |
11864 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Incorrect |
0 ms |
344 KB |
Output isn't correct |
2 |
Halted |
0 ms |
0 KB |
- |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
79 ms |
11740 KB |
Output is correct |
2 |
Correct |
86 ms |
11872 KB |
Output is correct |
3 |
Correct |
55 ms |
12168 KB |
Output is correct |
4 |
Correct |
52 ms |
12124 KB |
Output is correct |
5 |
Correct |
51 ms |
11864 KB |
Output is correct |
6 |
Correct |
51 ms |
11612 KB |
Output is correct |
7 |
Correct |
51 ms |
11604 KB |
Output is correct |
8 |
Correct |
52 ms |
11952 KB |
Output is correct |
9 |
Correct |
51 ms |
11848 KB |
Output is correct |
10 |
Correct |
62 ms |
11968 KB |
Output is correct |
11 |
Correct |
45 ms |
11864 KB |
Output is correct |
12 |
Correct |
47 ms |
12120 KB |
Output is correct |
13 |
Correct |
45 ms |
11864 KB |
Output is correct |
14 |
Correct |
45 ms |
11864 KB |
Output is correct |
15 |
Correct |
53 ms |
11828 KB |
Output is correct |
16 |
Correct |
50 ms |
11868 KB |
Output is correct |
17 |
Correct |
45 ms |
12124 KB |
Output is correct |
18 |
Correct |
52 ms |
12124 KB |
Output is correct |
19 |
Correct |
52 ms |
11612 KB |
Output is correct |
20 |
Correct |
45 ms |
11864 KB |
Output is correct |
21 |
Incorrect |
69 ms |
11864 KB |
Output isn't correct |
22 |
Halted |
0 ms |
0 KB |
- |