| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 369269 | Berted | Feast (NOI19_feast) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <iostream>
#include <queue>
#include <set>
#define ll long long
#define pii pair<int,int>
#define fst first
#define snd second
#define ppp pair<pair<ll,int>,pii>
using namespace std;
int n,k,l=-1;ll ar[300001]={},to=0;set<pii> s;
priority_queue<ppp> pq;
int main()
{
ios::sync_with_stdio(0);cin.tie(0);cout.tiie(0);
cin>>n>>k;
for (int i=1;i<=n;i++)
{
cin>>ar[i];
if (ar[i]>=0)
{
if (l==-1) {l = i;}
}
else
{
if (l!=-1) {to+=ar[i-1] - ar[l-1];s.insert({l,i-1});l=-1;}
}
ar[i] += ar[i-1];
}
if (l!=-1) {s.insert({l,n});to+=ar[n] - ar[l-1];}
l = -1;
for (auto dt:s)
{
pq.push({{-(ar[dt.snd] - ar[dt.fst-1]),0},{dt.fst,dt.snd}});
if (l!=-1)
{
pq.push({{ar[dt.fst-1] - ar[l],1},{l+1,dt.fst-1}});
}
l = dt.snd;
}
while (s.size()>k)
{
while (pq.size())
{
if (pq.top().fst.snd)
{
auto it = s.lower_bound({pq.top().snd.snd,pq.top().snd.snd});
pii dt = {-1,-1},dt2 = {-1,-1};
if (it!=s.begin()) {dt2 = *prev(it);}
if (it!=s.end()) {dt = *it;}
if (make_pair(dt2.snd+1,dt.fst-1)!=pq.top().snd) {pq.pop();}
else {break;}
}
else
{
auto it = s.lower_bound({pq.top().snd.fst,pq.top().snd.snd});
pii dt = {-1,-1};
if (it!=s.end()) {dt = *it;}
if (dt!=pq.top().snd) {pq.pop();}
else {break;}
}
}
to += pq.top().fst.fst;
if (pq.top().fst.snd)
{
auto it = s.lower_bound(pq.top().snd);
int l=prev(it)->fst,r=it->snd;
s.erase(prev(it));s.erase(it);pq.pop();
s.insert({l,r});
pq.push({{-(ar[r] - ar[l-1]),0},{l,r}});
}
else
{
auto it = s.lower_bound(pq.top().snd);
int l=pq.top().snd.fst,r=pq.top().snd.snd;
if (next(it)!=s.end()) {r = next(it)->fst-1;}
if (it!=s.begin()) {l = prev(it)->snd+1;}
s.erase(pq.top().snd);pq.pop();
pq.push({{ar[r] - ar[l-1],1},{l,r}});
}
}
cout<<to<<"\n";
return 0;
}