Submission #236199

#TimeUsernameProblemLanguageResultExecution timeMemory
236199MarcoMeijerWiring (IOI17_wiring)C++14
17 / 100
1089 ms7144 KiB
#include "wiring.h"
#include <bits/stdc++.h>
using namespace std;

//macros
typedef long long ll;
typedef pair<int, int> ii;
typedef tuple<int, int, int> iii;
typedef vector<int> vi;
typedef vector<ii> vii;
typedef vector<iii> viii;
typedef vector<ll> vll;
#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()


//===================//
//   begin program   //
//===================//

const int MX = 5e5;
vii all;
bitset<MX> connected;
int n, m, N;
ll nxtC[MX];
ll dp[MX];

int gd(int i, int j) {
    return abs(all[i].fi - all[j].fi);
}
ll getCost(int x, int y, int cent) {
    ll ans = 0;
    int R=cent-x, B=y-cent+1;
    if(R <= 0) return INF;
    if(B <= 0) return INF;
    if(R < B) {
        RE(i,R) ans += gd(x+i, x+R-1);
        RE(i,B) ans += gd(y-i, x+R-1);
    } else {
        RE(i,R) ans += gd(x+i, y-B+1);
        RE(i,B) ans += gd(y-i, y-B+1);
    } 
    return ans;
}

long long min_total_length(std::vector<int> r, std::vector<int> b) {
    n = r.sz, m = b.sz;
    N = n+m;
    {
        int i=0, j=0;
        while(i != n || j != m) {
            if(j == m || (i != n && r[i] < b[j])) {
                all.pb({r[i++], 0});
            } else {
                all.pb({b[j++], 1});
            }
        }
    }

    dp[0] = 0;
    int beg = -1, cent=-1, lastCol=-1;
    ll regionMin = INF;
    REP(i,1,N+1) {
        if(all[i-1].se != lastCol) {
            lastCol = all[i-1].se;
            beg = cent;
            cent = i-1;
            regionMin = INF;
        }
        dp[i] = INF;
        ll mn = regionMin;
        if(beg == -1) continue;
        REV(j,beg,cent+1) {
            mn = min(mn, dp[j]);
            ll cst = getCost(j,i-1,cent);
            dp[i] = min(dp[i], mn+cst);
        }
        regionMin = min(regionMin, dp[i]);
    }
	return dp[N];
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...