Submission #1257491

#	Time	Username	Problem	Language	Result	Execution time	Memory
1257491		ThylOne	Wiring (IOI17_wiring)	C++20	13 / 100	21 ms	4280 KiB

wiring

#include "wiring.h"
#include <bits/stdc++.h>

using namespace std;
long long min_total_length(std::vector<int> r, std::vector<int> b) {
	vector<pair<int, bool>> boxes;
	for(auto red:r)boxes.emplace_back(red,false);
	for(auto blue:b)boxes.emplace_back(blue, true);
	sort(boxes.begin(), boxes.end(), [](auto a, auto b){return a.first < b.first;});
	auto f = [&](){
		long long ans = 0;
		stack<pair<int,bool>> S;
		int last[2] = {-1,-1};
		for(int i = 0;i < boxes.size() ; i++){
			last[boxes[i].second] = boxes[i].first;
			if(S.empty()){
				S.push(boxes[i]);
			}else if(S.top().second != boxes[i].second){
				ans += abs(boxes[i].first - S.top().first);
				S.pop();
			}else{
				S.push(boxes[i]);
			}
		}
		while(!S.empty()){
			ans += abs(S.top().first - last[!S.top().second]);
			S.pop();
		}
		return ans;
	};
	long long ans = f();
	sort(boxes.begin(), boxes.end(), [](auto a, auto b){return a.first > b.first;});
	ans = min(ans, f());
	
	return ans;
}

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