Submission #120168

#	Time	Username	Problem	Language	Result	Execution time	Memory
120168		MvC	전선 연결 (IOI17_wiring)	C++11	30 / 100	1064 ms	45412 KiB

wiring

#pragma GCC target("avx2")
#pragma GCC optimization("O3")
#pragma GCC optimization("unroll-loops")
#include<bits/stdc++.h>
#include "wiring.h"
#define rc(x) return cout<<x<<endl,0
#define pb push_back
#define mkp make_pair
#define in insert
#define er erase
#define fd find
#define fr first
#define sc second
typedef long long ll;
typedef long double ld;
const ll INF=0x3f3f3f3f3f3f3f3f;
const ll llinf=(1LL<<62);
const int inf=(1<<30);
const ll nmax=2e5+50;
const int mod=1e9+7;
using namespace std;
int n,m,i,j,t,g;
ll x,y,cur;
vector<ll>ans[nmax],mn[nmax],suf[nmax],bst[nmax],gr[nmax];
vector<pair<ll,int> >a;
ll min_total_length(vector<int>R,vector<int>B)
{
	n=R.size(),m=B.size();
	for(i=1;i<=n;i++)
	{
		x=R[i-1];
		a.pb(mkp(x,0));
	}
	for(i=1;i<=m;i++)
	{
		x=B[i-1];
		a.pb(mkp(x,1));
	}
	sort(a.begin(),a.end());
	for(i=0;i<n+m;i++)
	{
		if(!i || a[i-1].sc!=a[i].sc)g++;
		gr[g].pb(a[i].fr);
	}
	for(i=1;i<=g;i++)
	{
		for(j=gr[i].size()-1,t=0;j>=0;j--,t++)
		{
			suf[i].pb(0);
			ans[i].pb(0);
			bst[i].pb(0);
			mn[i].pb(0);
			suf[i][t]=gr[i].back()-gr[i][j];
			if(t)suf[i][t]+=suf[i][t-1];
		}
		if(i>2)
		{
			x=0;
			for(j=0;j<gr[i].size();j++)
			{
				x+=gr[i][j]-gr[i][0];
				y=1LL*(j+1)*(gr[i][0]-gr[i-1].back());
				cur=x+y+bst[i-1][min(j,(int)gr[i-1].size()-1)];
				if(j+1<gr[i-1].size())cur=min(cur,x+mn[i-1][j+1]);
				ans[i][j]=cur;
			}
		}
		else if(i==2)
		{
			x=0;
			for(j=0;j<gr[i].size();j++)
			{
				x+=gr[i][j]-gr[i][0];
				y=1LL*max((int)gr[1].size(),j+1)*(gr[2][0]-gr[1].back());
				ans[i][j]=x+y+suf[1].back();
			}
		}
		if(i<g && i>1)
		{
			for(j=0;j<gr[i].size();j++)
			{
				x=0;
				for(t=j;t>=0;t--)
				{
					x+=gr[i+1][0]-gr[i][t];
					if(t)ans[i][j]=min(ans[i][j],ans[i][t-1]+x);
					else if(i!=2)ans[i][j]=min(ans[i][j],ans[i-1].back()+x);
				}
			}
			for(j=gr[i].size()-1,t=0;j>=0;j--,t++)
			{
				bst[i][t]=suf[i][t];
				if(j)bst[i][t]+=ans[i][j-1];
				else
				{
					if(i==2)bst[i][t]+=ans[i][0];
					else bst[i][t]+=ans[i-1].back();
				}
				if(t)bst[i][t]=min(bst[i][t],bst[i][t-1]);
				mn[i][j]=suf[i][j]+1LL*(j+1)*(gr[i+1][0]-gr[i].back());
				if(t)mn[i][j]+=ans[i][t-1];
				else
				{
					if(i==2)mn[i][j]+=ans[i][0];
					else mn[i][j]+=ans[i-1].back();
				}
				if(j<gr[i].size()-1)mn[i][j]=min(mn[i][j],mn[i][j+1]);
			}
		}
	}
	return ans[g].back();
}
/*int main()
{
	//freopen("sol.in","r",stdin);
	//freopen("sol.out","w",stdout);
	//mt19937 rng(chrono::steady_clock::now().time_since_epoch().count());
	ios_base::sync_with_stdio(false);cin.tie(0);cerr.tie(0);cout.tie(0);
	cin>>n>>m;
	for(i=1;i<=n;i++)
	{
		cin>>x;
		a.pb(mkp(x,0));
	}
	for(i=1;i<=m;i++)
	{
		cin>>x;
		a.pb(mkp(x,1));
	}
	sort(a.begin(),a.end());
	for(i=0;i<n+m;i++)
	{
		if(!i || a[i-1].sc!=a[i].sc)g++;
		gr[g].pb(a[i].fr);
	}
	for(i=1;i<=g;i++)
	{
		cout<<i<<" ";
		for(j=0;j<gr[i].size();j++)cout<<gr[i][j]<<" ";
		cout<<endl;
	}
	for(i=1;i<=g;i++)
	{
		cout<<"! "<<i<<endl;
		for(j=gr[i].size()-1,t=0;j>=0;j--,t++)
		{
			suf[i].pb(0);
			ans[i].pb(0);
			bst[i].pb(0);
			mn[i].pb(0);
			suf[i][t]=gr[i].back()-gr[i][j];
			if(t)suf[i][t]+=suf[i][t-1];
			//cout<<suf[i][t]<<" ";
		}
		//cout<<endl;
		if(i>2)
		{
			x=0;
			for(j=0;j<gr[i].size();j++)
			{
				x+=gr[i][j]-gr[i][0];
				y=1LL*(j+1)*(gr[i][0]-gr[i-1].back());
				cur=x+y+bst[i-1][min(j,(int)gr[i-1].size()-1)];
				if(j+1<gr[i-1].size())cur=min(cur,x+mn[i-1][j+1]);
				ans[i][j]=cur;
			}
		}
		else if(i==2)
		{
			x=0;
			for(j=0;j<gr[i].size();j++)
			{
				x+=gr[i][j]-gr[i][0];
				y=1LL*max((int)gr[1].size(),j+1)*(gr[2][0]-gr[1].back());
				ans[i][j]=x+y+suf[1].back();
			}
		}
		if(i<g && i>1)
		{
			for(j=0;j<gr[i].size();j++)
			{
				x=0;
				for(t=j;t>=0;t--)
				{
					x+=gr[i+1][0]-gr[i][t];
					if(t)ans[i][j]=min(ans[i][j],ans[i][t-1]+x);
					else ans[i][j]=min(ans[i][j],ans[i-1].back()+x);
				}
			}
			for(j=gr[i].size()-1,t=0;j>=0;j--,t++)
			{
				bst[i][t]=suf[i][t];
				if(j)bst[i][t]+=ans[i][j-1];
				else
				{
					if(i==2)bst[i][t]+=ans[i][0];
					else bst[i][t]+=ans[i-1].back();
				}
				if(t)bst[i][t]=min(bst[i][t],bst[i][t-1]);
				mn[i][j]=suf[i][j]+1LL*(j+1)*(gr[i+1][0]-gr[i].back());
				if(t)mn[i][j]+=ans[i][t-1];
				else
				{
					if(i==2)mn[i][j]+=ans[i][0];
					else mn[i][j]+=ans[i-1].back();
				}
				if(j<gr[i].size()-1)mn[i][j]=min(mn[i][j],mn[i][j+1]);
			}
		}
		for(j=0;j<gr[i].size();j++)cout<<ans[i][j]<<" ";
		cout<<endl;
	}
	cout<<ans[g].back()<<endl;
    return 0;
}
/*
4 5
1 2 3 7
0 4 5 9 10
*/

Compilation message (stderr)

wiring.cpp:2:0: warning: ignoring #pragma GCC optimization [-Wunknown-pragmas]
 #pragma GCC optimization("O3")
 
wiring.cpp:3:0: warning: ignoring #pragma GCC optimization [-Wunknown-pragmas]
 #pragma GCC optimization("unroll-loops")
 
wiring.cpp:216:1: warning: "/*" within comment [-Wcomment]
 /*
  
wiring.cpp: In function 'll min_total_length(std::vector<int>, std::vector<int>)':
wiring.cpp:59:13: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for(j=0;j<gr[i].size();j++)
            ~^~~~~~~~~~~~~
wiring.cpp:64:11: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(j+1<gr[i-1].size())cur=min(cur,x+mn[i-1][j+1]);
        ~~~^~~~~~~~~~~~~~~
wiring.cpp:71:13: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for(j=0;j<gr[i].size();j++)
            ~^~~~~~~~~~~~~
wiring.cpp:80:13: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    for(j=0;j<gr[i].size();j++)
            ~^~~~~~~~~~~~~
wiring.cpp:107:9: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     if(j<gr[i].size()-1)mn[i][j]=min(mn[i][j],mn[i][j+1]);
        ~^~~~~~~~~~~~~~~

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