| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 248731 | MarcoMeijer | 전선 연결 (IOI17_wiring) | C++14 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "wiring.h"
#include <bits/stdc++.h>
using namespace std;
//macros
typedef long long ll;
typedef pair<int, int> ii;
typedef pair<ll, ll> lll;
typedef tuple<int, int, int> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<iii> viii;
typedef vector<ll> vll;
typedef vector<lll> vlll;
#define REP(a,b,c) for(int a=int(b); a<int(c); a++)
#define RE(a,c) REP(a,0,c)
#define RE1(a,c) REP(a,1,c+1)
#define REI(a,b,c) REP(a,b,c+1)
#define REV(a,b,c) for(int a=int(c-1); a>=int(b); a--)
#define INF 1e18
#define pb push_back
#define fi first
#define se second
#define sz size()
const int MX = 1e6;
vi f[2];
ll N[2], n = 0;
ll a[MX], b[MX], sm[MX], ch[MX], CENT[MX];
ll dp[MX];
ll getCost(ll bg, ll ed) {
if(ch[ed] - ch[bg] != 1) return INF;
ll cent = CENT[ed];
ll cost = 0;
REP(i,bg ,cent) cost -= a[i];
REP(i,cent,ed+1) cost += a[i];
ll left = cent - bg;
ll right = ed+1 - cent;
if(left > right) {
cost += a[cent ]*(left - right);
} else {
cost -= a[cent-1]*(right - left);
}
return cost;
}
ll min_total_length(std::vector<int> R, std::vector<int> B) {
f[0] = R; f[1] = B;
RE(i,2) N[i]=f[i].sz;
// fill a and b
priority_queue<ii, vii, greater<ii>> pq;
RE(i,2) RE(j,N[i]) pq.push({f[i][j], i});
while(!pq.empty()) {
ii p = pq.top(); pq.pop();
a[n] = p.fi; b[n] = p.se;
n++;
}
sm[0] = 0; ch[0] = 0; CENT[0] = 0;
REP(i,1,n+1) {
ch[i] = ch[i-1];
sm[i] = sm[i-1]+a[i-1];
CENT[i] = CENT[i-1];
if(b[i] != b[i-1]) ch[i]++, CENT[i]=i;
}
// subt 2
if(getCost(0, n-1) != INF) return getCost(0, n-1);
dp[0] = 0;
REP(i,1,n+1) {
dp[i] = INF;
ll ed = i-1;
ll cent = CENT[ed];
ll beg = CENT[max(0ll, cent-1)];
if(b[i] != b[i-1]) {
REV(j,beg,cent) dp[j] = min(dp[j], dp[j+1]);
}
REV(j,0,i) {
if(ch[ed] - ch[bg] > 1) break;
dp[i] = min(dp[i], dp[j] + getCost(j,ed));
}
}
return dp[n];
}