Submission #389989

#	Time	Username	Problem	Language	Result	Execution time	Memory
389989		Keshi	Wiring (IOI17_wiring)	C++17	17 / 100	1089 ms	10164 KiB

wiring

//In the name of God
#include <bits/stdc++.h>
#include "wiring.h"
using namespace std;

typedef long long ll;
typedef pair<ll, ll> pll;

const ll maxn = 2e5 + 100;
const ll mod = 1e9 + 7;
const ll inf = 1e18;

#define fast_io ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define file_io freopen("input.txt", "r+", stdin);freopen("output.txt", "w+", stdout);
#define pb push_back
#define Mp make_pair
#define F first
#define S second
#define Sz(x) ll((x).size())
#define all(x) (x).begin(), (x).end()
#define lc (id << 1)
#define rc (lc | 1)

ll ls[maxn], dp[maxn];
vector<pll> a;

ll solve(ll k, ll j, ll i){
	ll x = max(i - j, j - k) * (a[j + 1].F - a[j].F);
	for(ll o = k + 1; o < j; o++){
		x += (a[o + 1].F - a[o].F) * (o - k);
	}
	for(ll o = j + 1; o < i; o++){
		x += (a[o + 1].F - a[o].F) * (i - o);
	}
	return x + min(dp[k + 1], dp[k + 2]);
}

ll get(ll i){
	ll j = ls[i];
	if(j == -1) return inf;
	ll k = ls[j];
	ll x = inf;
	for(ll o = k; o < j; o++){
		x = min(x, solve(o, j, i));
	}
	return x;
}

long long min_total_length(vector<int> r, vector<int> b){
	for(ll i : r){
		a.pb(Mp(i, 0));
	}
	for(ll j : b){
		a.pb(Mp(j, 1));
	}
	sort(all(a));
	ll las[2] = {-1, -1};
	ll n = Sz(a);
	for(ll i = 0; i < n; i++){
		ls[i] = las[a[i].S ^ 1];
		las[a[i].S] = i;
		ll j = ls[i];
		dp[i + 1] = get(i);
		continue;
		if(j == -1) continue;
		if(i - j > 2){
			dp[i + 1] = dp[i] + a[i].F - a[j].F;
			continue;
		}
		if(i - j == 1){
			ll x = 0;
			for(ll k = j; k > ls[j]; k--){
				x += a[i].F - a[k].F;
				dp[i + 1] = min(dp[i + 1], dp[k] + x);
			}
			continue;
		}
		ll x = a[i].F - a[j].F;
		dp[i + 1] = x + a[i - 1].F - a[j].F + dp[j];
		for(ll k = j - 1; k > ls[j]; k--){
			x += a[i - 1].F - a[k].F;
			dp[i + 1] = min(dp[i + 1], dp[k] + x);
		}
	}

	return dp[n];
}

/*int main(){
    fast_io;



    return 0;
}*/

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