제출 #363003

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
363003		Seanliu	Ideal city (IOI12_city)	C++14	32 / 100	196 ms	876 KiB

city

#include <vector>
#include <algorithm>
#include <iostream>
#include <queue>
#include <utility>
#define pii pair<int,int>
#define F first
#define S second
using namespace std;
 
const int MOD = 1e9, maxN = 1e5 + 326;
const long long int INF = (1LL << 31) - 2;
 
int mp[2020][2020], vis[maxN], d[maxN], dir[4][2] = {{1, 0}, {0, 1}, {-1, 0}, {0, -1}};
queue<int> que;
vector<pii> pos;
 
inline int add(int a, int b){
	return (a + b >= MOD ? a + b - MOD : a + b);
}

struct BIT{
	int arr[maxN];
	BIT(){}
	void modify(int p, int x){
		for(; p < maxN; p += (p & -p)) arr[p] = add(arr[p], x);
	}
	int sum(int p){
		int r = 0;
		for(; p; p -= (p & -p)) r = add(r, arr[p]);
		return r;
	}
} cnt, xy, xy2;
 
int DistanceSum(int N, int *X, int *Y) {
	if(N <= 2000){
		int minX = INF, minY = INF;
		for(int i = 0; i < N; i++){
			minX = min(minX, X[i]);
			minY = min(minY, Y[i]);
		}
		for(int i = 0; i < N; i++){
			X[i] -= minX - 1;
			Y[i] -= minY - 1;
			mp[X[i]][Y[i]] = i + 1;
		}
		long long int ans = 0;
		for(int i = 0; i < N; i++){
			fill(d, d + N, INF);	
			fill(vis, vis + N, false);
			d[i] = 0;
			que.push(i);
			vis[i] = true;
			//cout << "i = " << i << endl;
			while(que.size()){
				int id = que.front(); que.pop();
				for(int j = 0; j < 4; j++){
					if(mp[X[id] + dir[j][0]][Y[id] + dir[j][1]]){
						int nid = mp[X[id] + dir[j][0]][Y[id] + dir[j][1]] - 1;
						if(!vis[nid]){
							d[nid] = d[id] + 1;
							//cout << "d[" << i << "][" << nid << "] = " << d[nid] << endl;
							ans = add(ans, d[nid]);
							vis[nid] = true;
							que.push(nid);
						}
					}
				}
			}
		}
		return ans / 2;
	}
	int minX = INF, minY = INF;
	for(int i = 0; i < N; i++){
		minX = min(minX, X[i]);
		minY = min(minY, Y[i]);
	}
	for(int i = 0; i < N; i++){
		X[i] -= minX - 1;
		Y[i] -= minY - 1;
		pos.emplace_back(X[i], Y[i]);
	}
	sort(pos.begin(), pos.end());	
	long long int ans = 0;
	//xy: x + y
	//xy2: y - x 
	for(auto f : pos){
		int x = f.F, y = f.S;
		ans = add(ans, (MOD - xy.sum(y))% MOD);
		ans = add(ans, (long long int)cnt.sum(y) * (long long int)add(x, y) % MOD);

		ans = add(ans, (xy2.sum(maxN - 1) - xy2.sum(y) + MOD) % MOD);
		ans = add(ans, (long long int)((cnt.sum(maxN - 1) - cnt.sum(y) + MOD) % MOD) * (long long int)((x - y + MOD) % MOD));

		cnt.modify(y, 1);
		xy.modify(y, add(x, y));
		xy2.modify(y, (y - x + MOD) % MOD);
	}
	return add(0, ans);
}

/*
int DistanceSum(int N, int *X, int *Y) {
		if(N > maxN) return 0;
		int minX = INF, minY = INF;
		for(int i = 0; i < N; i++){
			minX = min(minX, X[i]);
			minY = min(minY, Y[i]);
		}
		for(int i = 0; i < N; i++){
			X[i] -= minX - 1;
			Y[i] -= minY - 1;
			mp[X[i]][Y[i]] = i + 1;
		}
		long long int ans = 0;
		for(int i = 0; i < N; i++){
			fill(d, d + N, INF);	
			fill(vis, vis + N, false);
			d[i] = 0;
			que.push(i);
			vis[i] = true;
			//cout << "i = " << i << endl;
			while(que.size()){
				int id = que.front(); que.pop();
				for(int j = 0; j < 4; j++){
					if(mp[X[id] + dir[j][0]][Y[id] + dir[j][1]]){
						int nid = mp[X[id] + dir[j][0]][Y[id] + dir[j][1]] - 1;
						if(!vis[nid]){
							d[nid] = d[id] + 1;
							//cout << "d[" << i << "][" << nid << "] = " << d[nid] << endl;
							ans = add(ans, d[nid]);
							vis[nid] = true;
							que.push(nid);
						}
					}
				}
			}
		}
		return ans / 2;
	return 5;
	int minX = INF, minY = INF;
	for(int i = 0; i < N; i++){
		minX = min(minX, X[i]);
		minY = min(minY, Y[i]);
	}
	for(int i = 0; i < N; i++){
		X[i] -= minX - 1;
		Y[i] -= minY - 1;
		pos.emplace_back(X[i], Y[i]);
	}
	sort(pos.begin(), pos.end());	
	long long int ans = 0;
	for(auto f : pos){
		int x = f.F, y = f.S;
		ans = add(ans, MOD - xy.sum(y));
		ans = add(ans, cnt.sum(y) * add(x, y) % MOD);
		cnt.modify(y, 1);
		xy.modify(y, add(x, y));
	}
	return add(0, ans);
}
*/

/*
int DistanceSum(int N, int *X, int *Y) {
	if(N > maxN) return 0;
	int minX = INF, minY = INF;
	for(int i = 0; i < N; i++){
		minX = min(minX, X[i]);
		minY = min(minY, Y[i]);
	}
	for(int i = 0; i < N; i++){
		X[i] -= minX - 1;
		Y[i] -= minY - 1;
		mp[X[i]][Y[i]] = i + 1;
	}
	long long int ans = 0;
	for(int i = 0; i < N; i++){
		fill(d, d + N, INF);	
		fill(vis, vis + N, false);
		d[i] = 0;
		que.push(i);
		vis[i] = true;
		//cout << "i = " << i << endl;
		while(que.size()){
			int id = que.front(); que.pop();
			for(int j = 0; j < 4; j++){
				if(mp[X[id] + dir[j][0]][Y[id] + dir[j][1]]){
					int nid = mp[X[id] + dir[j][0]][Y[id] + dir[j][1]] - 1;
					if(!vis[nid]){
						d[nid] = d[id] + 1;
						//cout << "d[" << i << "][" << nid << "] = " << d[nid] << endl;
						ans = add(ans, d[nid]);
						vis[nid] = true;
						que.push(nid);
					}
				}
			}
		}
	}
	return ans / 2;
}
*/

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