Submission #1202270

#TimeUsernameProblemLanguageResultExecution timeMemory
1202270MongHwa전선 연결 (IOI17_wiring)C++20
Compilation error
0 ms0 KiB
#pragma GCC optimize ("Ofast")
#pragma GCC optimize ("unroll-loops")
#include <iostream>
#include <vector>
#include <cstdio>
#include <cassert>
using namespace std;

#define INF 0x7f7f7f7f
bool chk[100005];

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

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

		if(lval < rval)
		{
			ans += lval;
			chk[lpos] = 1;
		}
		else if(lval > rval)
		{
			ans += rval;
			chk[rpos] = 1;
		}
		else
		{
			ans += lval;
			if(!chk[lpos] && chk[rpos])
				chk[lpos] = 1;
			else if(chk[lpos] && !chk[rpos])
				chk[rpos] = 1;
			else
				chk[lpos] = 1;
		}
	}

	for(int i = 0; i < m; i++)
	{
		if(chk[i])
			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;
}


int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);

    int n, m;
	assert(2 == scanf("%d %d", &n, &m));

	vector<int> r(n), b(m);
	for(int i = 0; i < n; i++)
		assert(1 == scanf("%d", &r[i]));
	for(int i = 0; i < m; i++)
		assert(1 == scanf("%d", &b[i]));

	long long res = min_total_length(r, b);
	printf("%lld\n", res);

	return 0;
}

Compilation message (stderr)

/usr/bin/ld: /tmp/ccLiY38M.o: in function `main':
grader.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccb8Wm0s.o:wiring.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status