제출 #283326

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
283326		DystoriaX	전선 연결 (IOI17_wiring)	C++14	0 / 100	1 ms	384 KiB

wiring

#include "wiring.h"
#include <algorithm>
#include <iostream>
#include <set>

using namespace std;

vector<pair<int, int> > v;
const long long INF = 1e18;
long long dp[200010], val[200010];
multiset<long long> s;

long long min_total_length(std::vector<int> r, std::vector<int> b) {
	for(auto &k : r) v.emplace_back(k, 0);
	for(auto &k : b) v.emplace_back(k, 1);

	sort(v.begin(), v.end());

	int last = -1;
	for(int i = 1; i < (int) v.size(); i++){
		if(v[i].second == v[i - 1].second) continue;
		else {last = i; break;}
	}

	for(int i = 0; i < (int) v.size(); i++) dp[i] = 1e17;

	// cerr << last << "\n";
	int lpos = v[last - 1].first, cpos = v[last].first;
	int ls = -1, conmax = 0;

	for(int i = last; i < (int) v.size(); i++){
		if(v[i].second == v[i - 1].second){
			if(i - ls < conmax){
				s.erase(s.find(val[i - ls])), s.insert(val[i - ls] + cpos - lpos);
				dp[i] = *s.begin() + v[i].first - cpos;
			} else {
				dp[i] = dp[i - 1] + v[i].first - lpos;
			}
			
		} else {
			lpos = v[i - 1].first;
			cpos = v[i].first;
			ls = i;

			long long cost = 0;
			s.clear();
			for(int j = i - 1; j >= 0; j--){
				if(v[j].second == v[i].second) break;
				cost += v[i].first - v[j].first;

				long long pen = (j == 0) ? 0 : dp[j - 1];

				val[i - j] = pen + cost;
				s.insert(val[i - j]);
				conmax = i - j;
			}

			dp[i] = *s.begin();
		}

		// cerr << v[i].first << " " << i << " = " << dp[i] << "\n";
	}

	return dp[v.size() - 1];
}

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