제출 #142815

#제출 시각아이디문제언어결과실행 시간메모리
142815babo전선 연결 (IOI17_wiring)C++14
13 / 100
78 ms29048 KiB
#include <bits/stdc++.h>
#define inf 2000000000
#define all(x) (x).begin(),(x).end()
#define L long long

using namespace std;

struct S{
	L loc,det,chk,lev;
}a[200020];

set<L>st[200020];
L n,m;

L min_total_length(vector<int>r, vector<int>b){
	n=r.size();
	m=b.size();
	L i,ret=0;
	for(i=0;i<n;i++)
	{
		a[i]=(S){r[i],1,0};
	}
	for(i=n;i<n+m;i++)
	{
		a[i]=(S){b[i-n],2,0};
	}
	sort(a,a+n+m,[](S a,S b){
		return a.loc<b.loc;
	});

	L lev=100010;
	for(i=0;i<n+m;i++)
	{
		a[i].lev=lev;
		st[lev].insert(i);
		if(a[i].det==1&&a[i+1].det==1) lev++;
		if(a[i].det==2&&a[i+1].det==2) lev--;
		//printf("%lld %lld\n",a[i].det,a[i].loc);
	}
	for(i=0;i<n+m;i++)
	{
		if(a[i].chk) continue;
		set<L>::iterator it=st[a[i].lev].upper_bound(i);
		if(it!=st[a[i].lev].end())
		{
			//printf("match %lld %lld\n",i,*it);
			a[i].chk=1;
			a[*it].chk=1;
			ret+=a[*it].loc-a[i].loc;
			st[a[i].lev].erase(it);
			st[a[i].lev].erase(i);
		}
		else
		{
			L mi=inf;
			vector<int>::iterator it;
			if(a[i].det==1)
			{
				it=lower_bound(all(b),a[i].loc);
				if(it!=b.end())
					mi=min(mi,*it-a[i].loc);
				it=upper_bound(all(b),a[i].loc);
				if(it!=b.begin())
					mi=min(mi,a[i].loc-*(prev(it)));
			}
			else
			{
				it=lower_bound(all(r),a[i].loc);
				if(it!=r.end())
					mi=min(mi,*it-a[i].loc);
				it=upper_bound(all(r),a[i].loc);
				if(it!=r.begin())
					mi=min(mi,a[i].loc-*(prev(it)));
			}
			ret+=mi;
		}
	}
	return ret;
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...