# | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
---|---|---|---|---|---|---|---|
654962 | onlk97 | Candies (JOI18_candies) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include <bits/stdc++.h>
#define int long long
using namespace std;
int main(){
ios_base::sync_with_stdio(0); cin.tie(0);
int n;
cin>>n;
long long a[n+2];
a[0]=a[n+1]=-1e18;
for (int i=1; i<=n; i++) cin>>a[i];
long long l[n+1],r[n+1];
for (int i=1; i<=n; i++){
l[i]=i-1;
r[i]=i+1;
}
l[0]=0; r[n+1]=n+1;
bool used[n+2];
for (int i=0; i<=n+1; i++) used[i]=0;
set <pair <long long,int>,greater <pair <long long,int> > > s;
for (int i=1; i<=n; i++) s.insert(make_pair(a[i],i));
long long ans=0;
for (int iters=1; iters<=(n+1)/2; iters++){
die:;
pair <long long,int> tp=*s.begin(); s.erase(s.begin());
if (used[tp.second]) goto die;
ans+=tp.first;
cout<<ans<<'\n';
int ll=l[tp.second],rr=r[tp.second];
a[tp.second]=a[ll]+a[rr]-a[tp.second];
l[tp.second]=l[ll];
r[tp.second]=r[rr];
used[ll]=1;
used[rr]=1;
r[l[tp.second]]=tp.second;
l[r[tp.second]]=tp.second;
if (ll>=1&&rr<=n) s.insert(make_pair(a[tp.second],tp.second));
}
}