10560 | 2024-04-05 13:32:10 | 111 | Az IKPC legerősebb csapata | cpp17 | Wrong answer 0/100 | 114ms | 10704 KiB |
#include<bits/stdc++.h>
using namespace std;
#define int long long
#define pii pair<int,int>
signed main(){
ios::sync_with_stdio(false);
cin.tie(nullptr);
int N,K;
cin>>N>>K;
vector<int>v(N+1),w(N+1);
for(int i=1;i<=N;i++){
cin>>v[i];
}
for(int i=1;i<=N;i++){
cin>>w[i];
}
vector<vector<int>>g(N+1);
vector<int>s;
for(int i=N;i>=1;i--){
while(!s.empty()&&v[s.back()]<=v[i]){
s.pop_back();
}
g[s.empty()?0:s.back()].push_back(i);
s.push_back(i);
}
auto solveMin=[&](int y)->pii{
auto dfs=[&](auto self,int i)->pair<pii,pii>{
pii a{},b{};
for(int j:g[i]){
auto [p,q]=self(self,j);
a.first+=p.first;
a.second+=p.second;
if(q.first-p.first>b.first){
b.first=q.first-p.first;
b.second=q.second-p.second;
}
}
b.first+=a.first+w[i];
b.second+=a.second;
if(w[i]>y){
a.first+=w[i]-y;
a.second++;
}
if(b.first-y>a.first){
a.first=b.first-y;
a.second=b.second+1;
}
return{a,b};
};
return dfs(dfs,0).first;
};
auto solveMax=[&](int y)->pii{
auto dfs=[&](auto self,int i)->pair<pii,pii>{
pii a{},b{};
for(int j:g[i]){
auto [p,q]=self(self,j);
a.first+=p.first;
a.second+=p.second;
if(q.first-p.first>b.first||q.first-p.first==b.first&&q.second-p.second>b.second){
b.first=q.first-p.first;
b.second=q.second-p.second;
}
}
b.first+=a.first+w[i];
b.second+=a.second;
if(w[i]>=y){
a.first+=w[i]-y;
a.second++;
}
if(b.first-y>=a.first){
a.first=b.first-y;
a.second=b.second+1;
}
return{a,b};
};
return dfs(dfs,0).first;
};
int l=0,h=1e16;
while(l<h){
int m=(l+h)/2;
auto[x,c]=solveMin(m);
if(c<=K){
h=m;
}
else{
l=m+1;
}
}
auto[x,c]=solveMax(h);
cout<<x+min(c,K)*h<<'\n';
return 0;
}
Subtask | Sum | Test | Verdict | Time | Memory | ||
---|---|---|---|---|---|---|---|
subtask1 | 0/0 | ||||||
1 | Accepted | 3ms | 1696 KiB | ||||
2 | Accepted | 3ms | 1928 KiB | ||||
subtask2 | 0/9 | ||||||
3 | Accepted | 3ms | 2020 KiB | ||||
4 | Wrong answer | 3ms | 2140 KiB | ||||
5 | Wrong answer | 3ms | 2360 KiB | ||||
6 | Accepted | 3ms | 2568 KiB | ||||
7 | Accepted | 3ms | 2588 KiB | ||||
8 | Wrong answer | 2ms | 2580 KiB | ||||
9 | Wrong answer | 3ms | 2708 KiB | ||||
10 | Accepted | 3ms | 3068 KiB | ||||
11 | Wrong answer | 2ms | 2944 KiB | ||||
12 | Wrong answer | 3ms | 3176 KiB | ||||
subtask3 | 0/7 | ||||||
13 | Accepted | 3ms | 3304 KiB | ||||
14 | Accepted | 4ms | 3488 KiB | ||||
15 | Wrong answer | 6ms | 3588 KiB | ||||
16 | Accepted | 6ms | 3768 KiB | ||||
17 | Accepted | 7ms | 4068 KiB | ||||
18 | Accepted | 7ms | 3976 KiB | ||||
19 | Accepted | 7ms | 4288 KiB | ||||
20 | Accepted | 7ms | 4452 KiB | ||||
21 | Accepted | 8ms | 4292 KiB | ||||
subtask4 | 0/11 | ||||||
22 | Accepted | 93ms | 9624 KiB | ||||
23 | Accepted | 90ms | 9640 KiB | ||||
24 | Accepted | 93ms | 9872 KiB | ||||
25 | Accepted | 92ms | 9932 KiB | ||||
26 | Wrong answer | 101ms | 9932 KiB | ||||
27 | Accepted | 105ms | 10136 KiB | ||||
subtask5 | 0/22 | ||||||
28 | Accepted | 7ms | 4836 KiB | ||||
29 | Accepted | 7ms | 4836 KiB | ||||
30 | Accepted | 7ms | 4832 KiB | ||||
31 | Wrong answer | 7ms | 4816 KiB | ||||
32 | Accepted | 7ms | 5108 KiB | ||||
33 | Wrong answer | 8ms | 5152 KiB | ||||
34 | Wrong answer | 6ms | 5084 KiB | ||||
subtask6 | 0/51 | ||||||
35 | Accepted | 103ms | 10348 KiB | ||||
36 | Accepted | 103ms | 10468 KiB | ||||
37 | Wrong answer | 104ms | 10348 KiB | ||||
38 | Accepted | 108ms | 10368 KiB | ||||
39 | Accepted | 108ms | 10472 KiB | ||||
40 | Accepted | 114ms | 10456 KiB | ||||
41 | Wrong answer | 83ms | 10460 KiB | ||||
42 | Accepted | 96ms | 10704 KiB | ||||
43 | Accepted | 105ms | 10584 KiB | ||||
44 | Accepted | 97ms | 10672 KiB | ||||
45 | Accepted | 96ms | 10672 KiB |