Submission #1202427

#	Time	Username	Problem	Language	Result	Execution time	Memory
1202427		MongHwa	Wiring (IOI17_wiring)	C++20	13 / 100	37 ms	6568 KiB

wiring

#include "wiring.h"
#include <set>
using namespace std;

#define INF 0x7f7f7f7f
bool chk[100005];

long long min_total_length(std::vector<int> r, std::vector<int> b) {
	if(r.size() > b.size())
		swap(r, b);
	
	int n = (int)r.size(), m = (int)b.size();

	set<int> st;
	for(int i = 0; i < m; i++)
		st.insert(b[i]);
	
	long long ans = 0;
	for(int i = 0; i < n; i++)
	{
		auto it = st.lower_bound(r[i]);
		int lval = INF, rval = INF;
		if(it != st.end())
			rval = *it-r[i];
		if(it != st.begin())
		{
			it = prev(it);
			lval = r[i]-*it;
		}

		ans += min(lval, rval);
		if(lval <= rval)
			st.erase(-lval+r[i]);
		else
			st.erase(rval+r[i]);
	}
	
	for(int i = 0; i < m; i++)
	{
		if(st.find(b[i]) == st.end())
			continue;
		
		int lval = INF, rval = INF;
		auto it = lower_bound(r.begin(), r.end(), b[i]);
		if(it != r.end())
			rval = *it-b[i];
		if(it != r.begin())
		{
			it = prev(it);
			lval = b[i]-*it;
		}

		ans += min(lval, rval);
	}

	return ans;
}

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