Submission #363096

#	Time	Username	Problem	Language	Result	Execution time	Memory
363096		Kevin_Zhang_TW	Ideal city (IOI12_city)	C++17	32 / 100	154 ms	7660 KiB

city

#include <bits/stdc++.h>
using namespace std;
using ll = long long;
#define pb emplace_back
#define AI(i) begin(i), end(i)
template<class T> bool chmin(T &a, T b) { return b < a && (a = b, true); }
template<class T> bool chmax(T &a, T b) { return a < b && (a = b, true); }
#ifdef KEV
#define DE(args...) kout("[ " + string(#args) + " ] = ", args)
void kout() { cerr << endl; }
template<class T, class ...U> void kout(T a, U ...b) { cerr << a << ' ', kout(b...); }
template<class T> void debug(T L, T R) { while (L != R) cerr << *L << " \n"[next(L) == R], ++L; }
#else
#define DE(...) 0
#define debug(...) 0
#endif
const int MAX_N = 300010, inf = 1e9;

int N;
vector<int> edge[MAX_N];
ll bfs_all(int s) {
	vector<int> dis(N, inf); dis[s] = 0;
	queue<int> q; q.push(s);
	ll res = 0;
	while (q.size()) {
		int x = q.front(); q.pop();
		res += dis[x];
		for (int u : edge[x]) if (chmin(dis[u], dis[x] + 1))
			q.push(u);
	}
	return res;
}
ll cal(int N, int *X, int *Y) {
	ll res = 0;
	map<int, int> cnt;
	for (int i = 0;i < N;++i) ++cnt[X[i]];
	ll L = 0, R = N;
	for (auto [_, c] : cnt) {
		L += c, R -= c;
		res += L * R;
	}
	return res;
}
int DistanceSum(int N, int *X, int *Y) {
	::N = N;
	ll res = 0;
	if (N > 2000) {
		res = cal(N, X, Y) + cal(N, Y, X);
	}
	else {
		for (int i = 0;i < N;++i)
			for (int j = 0;j < N;++j) if (i != j) 
				if (abs(X[i]-X[j]) + abs(Y[i]-Y[j]) == 1)
					edge[i].pb(j);
		for (int i = 0;i < N;++i)
			res += bfs_all(i);
	}
	return res / 2 % 1'000'000'000;
}

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