Submission #258498

#	Time	Username	Problem	Language	Result	Execution time	Memory
258498		b00n0rp	Wiring (IOI17_wiring)	C++17	17 / 100	1093 ms	9576 KiB

wiring

#include "wiring.h"
#include<bits/stdc++.h>
using namespace std;

#define ll long long
#define vi vector<int>
#define pb push_back
#define pii pair<int,int>
#define F first
#define S second
#define REP(i,n) for(int i = 0; i < n; i++)
#define FOR(i,a,b) for(int i = a; i < b; i ++)
#define all(v) v.begin(),v.end()

vector<pii> a;
int n;
int comp[200005];
pii range[200005];
ll dp[200005],pref[200005],suff[200005];

const ll INF = 1e14;

ll min_total_length(vi r, vi b) {
	for(auto x:r) a.pb({x,0});
	for(auto x:b) a.pb({x,1});
	sort(all(a));
	n = a.size();
	int last = 0,cmp = 0;
	FOR(i,1,n){
		if(a[i].S != a[last].S){
			range[cmp] = {last,i-1};
			last = i;
			cmp++;
		}
		comp[i] = cmp;
	}
	range[cmp] = {last,n-1};
	pref[0] = 0;
	FOR(i,1,n){
		if(comp[i] == comp[i-1]){
			pref[i] = pref[i-1]+(a[i].F-a[range[comp[i]].F].F);
		}
	}
	suff[n-1] = 0;
	for(int i = n-2; i >= 0; i--){
		if(comp[i] == comp[i+1]){
			suff[i] = suff[i+1]+(a[range[comp[i]].S].F-a[i].F);
		}
	}
	REP(i,n){
		if(!comp[i]){
			dp[i] = INF;
			continue;
		}
		int st = range[comp[i]].F;
		int l1 = range[comp[st-1]].F,r2 = range[comp[st-1]].S;
		int diff = a[st].F-a[r2].F;
		dp[i] = INF;
		int r1 = max(2*st-i-1,l1-1);
		int l2 = r1+1;
		FOR(j,l1,r1+1){
			dp[i] = min(dp[i],pref[i]+suff[j]+diff*(st-j)+(j?min(dp[j-1],dp[j]):0));
		}
		FOR(j,l2,r2+1){
			dp[i] = min(dp[i],pref[i]+suff[j]+diff*(i-st+1)+(j?min(dp[j-1],dp[j]):0));
		}
		// cout << i << " " << a[i].F << " " << pref[i] << " " << suff[i] << " " << dp[i] << endl;
	}
	return dp[n-1];
}

• Subtask 1: 7 / 7
• Subtask 2: 0 / 13
• Subtask 3: 10 / 10
• Subtask 4: 0 / 25
• Subtask 5: 0 / 45