Submission #66041

#	Time	Username	Problem	Language	Result	Execution time	Memory
66041		mhnd	Ideal city (IOI12_city)	C++14	55 / 100	263 ms	32592 KiB

city

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

using namespace std;

typedef long long ll;
typedef unsigned long long ull;
const int N = 5e5+50;
const ll oo = 1e18;
const ll mod = 1e9;

int cost[2010][2010],r[2010][2010];
ll ans = 0;
int cur;
int dx[] = {1,-1,0,0};
int dy[] = {0,0,1,-1};

void bfs(pair<int,int> st){
	queue<pair<int,int>> q;
	q.push(st);
	int fr = r[st.first][st.second];
	cost[fr][fr] = 0;
	while(q.size()){
		pair<int,int> u = q.front();
		q.pop();
		int bf  = r[u.first][u.second];
		if(fr < bf)ans += cost[fr][bf];
		for(int i=0;i<4;i++){
			pair<int,int> v = {u.first+dx[i],u.second+dy[i]};
			int idx = r[v.first][v.second];
			if(idx==-1 || cost[fr][idx]!=-1)continue;
			cost[fr][idx] = cost[fr][bf]+1;
			q.push(v);
		}
	}
}

int DistanceSum(int N, int *X, int *Y) {
	if(N<=2000){
    	int mn1 = *min_element(X, X+N), mn2 = *min_element(Y, Y+N);
    	mn1-=5;
    	mn2-=5;
		memset(cost,-1,sizeof(cost));
		memset(r,-1,sizeof(r));
		for(int i=0;i<N;i++){
			X[i]-=mn1;
			Y[i]-=mn2;
			r[X[i]][Y[i]]=i;
		}
		for(int i=0;i<N;i++)bfs({X[i],Y[i]});
		return ans;
	}
	sort(X,X+N);
	int sum=0,ret=0,lst=X[0];
	for(int i=0;i<N;i++){
		if(X[i]!=lst){
			ret = (ret + (X[i]-lst)*1ll*(sum*1ll*(N-sum))%mod)%mod;
			lst = X[i];
		}
		sum++;
	}
	ans = ret;
	sort(Y,Y+N);
	sum=0;ret=0;lst=Y[0];
	for(int i=0;i<N;i++){
		if(Y[i]!=lst){
			ret = (ret + (Y[i]-lst)*1ll*(sum*1ll*(N-sum))%mod)%mod;
			lst = Y[i];
		}
		sum++;
	}
	ans = (ans+ret)%mod;
	return ans;
}

• Subtask 1: 11 / 11
• Subtask 2: 21 / 21
• Subtask 3: 23 / 23
• Subtask 4: 0 / 45