Submission #316837

#	Time	Username	Problem	Language	Result	Execution time	Memory
316837		amunduzbaev	Ideal city (IOI12_city)	C++14	32 / 100	168 ms	16248 KiB

city

//#include "grader.cpp"
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
const int N=2005, mod= 1e9;
int n, dist[N][N];
vector<vector<int>> edges(2005);

void Dijkstra(int cur){
	queue<int>q;
	q.push(cur);
	while(!q.empty()){
		int u=q.front(); q.pop();
		for(auto x:edges[u]){
			if(dist[cur][u]+1 < dist[cur][x]){
			dist[cur][x] = dist[cur][u]+1;
			q.push(x);
			}
		}
	}
}
const int NAX = 1e5+5;
int sufx[NAX], sufy[NAX];

int DistanceSum(int N, int *x, int *y) {
	n=N;
	if(n<=2000){
		for(int i=0;i<n;i++){
			for(int j=i+1;j<n;j++){
				dist[i][j]=2e9;
				dist[j][i]=2e9;
				if(abs(x[i]-x[j]) + abs(y[i]-y[j]) == 1){
					edges[i].pb(j);
					edges[j].pb(i);
				}
			}
		}

		for(int i=0;i<n;i++)
			Dijkstra(i);

		int ans=0;
		for(int i=0; i<n; i++){
			for(int j=i+1; j<n; j++){
				ans += dist[i][j];
				ans %= mod;
			}
		}
 		return ans;
	}
	else{
		sort(x,x+n);
		sort(y,y+n);
		sufx[n-1]=x[n-1], sufy[n-1]=y[n-1];
		for(int i=n-2; i>=0; i--){
			sufx[i] += sufx[i+1] + x[i];
			sufy[i] += sufy[i+1] + y[i];
		}
		int ans=0;
		for(int i=0;i<n;i++){
			int cur=x[i]*(n-i);
			ans+=(sufx[i]-cur);
			cur=y[i]*(n-1);
			ans+=(sufy[i]-cur);
		}
		return ans;

	}
}

/*

11
2 5
2 6
3 3
3 6
4 3
4 4
4 5
4 6
5 3
5 4
5 6

*/

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