Submission #1241372

#TimeUsernameProblemLanguageResultExecution timeMemory
1241372Dan4LifeWiring (IOI17_wiring)C++17
0 / 100
0 ms328 KiB
#include "wiring.h"
#include <bits/stdc++.h>
using namespace std;

#define pb push_back
#define sz(a) (int)a.size()
#define all(a) begin(a),end(a)

using ll = long long;
using vi = vector<int>;
using ar2 = array<ll,2>;

const int mxN = (int)4e5+10;
const ll LINF = (ll)4e18;
int n, m;
ll dp[mxN], lst[mxN], pref[2][mxN];
vector<ar2> v;

ll F(int l, int r){ return dp[l]+abs(pref[1][r+1]-pref[1][l] - (pref[0][r+1]-pref[0][l])); }

ll min_total_length(vi r, vi b) {
	n = sz(r), m = sz(b);
	for(auto u : r) v.pb({u,0});
	for(auto u : b) v.pb({u,1});
	sort(all(v));
	int cur[2]; cur[0]=cur[1]=-1;
	for(int i = 0; i < n+m; i++){
		lst[i] = cur[!v[i][1]];
		cur[v[i][1]]=i, dp[i+1] = LINF;
		for(int j : {0,1}) pref[j][i+1] = pref[j][i];
		pref[v[i][1]][i+1]+=v[i][0];
	}
	for(int i = 0; i < n+m; i++){
		int j = lst[i]; if(j<0) continue;
		int k = lst[j]+1;
		dp[i+1] = min(dp[i+1], dp[i]+v[i][0]-v[j][0]);
		if(j-k+1 >= i-j) dp[i+1]=min(dp[i+1],F(j+1-i+j,i));
	}
	return dp[n+m];
}
#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...