# |
Submission time |
Handle |
Problem |
Language |
Result |
Execution time |
Memory |
670237 |
2022-12-08T11:08:45 Z |
1bin |
Ideal city (IOI12_city) |
C++17 |
|
0 ms |
0 KB |
#include <bits/stdc++.h>
using namespace std;
#define all(v) v.begin(), v.end()
typedef long long ll;
const int NMAX = 1e5 + 5;
const ll mod = 1e9;
int n, X[NMAX], Y[NMAX], a[NMAX], w[NMAX];
vector<pair<int, int>> v;
vector<ll> adj[NMAX];
ll dfs(int x, int p){
ll ret = 0;
for(ll& nx : adj[x]){
if(nx == p) continue;
ret += dfs(nx, x);
ret += 1LL * w[nx] * (n - w[nx]); ret %= mod;
w[x] += w[nx];
}
return ret;
}
ll go(){
sort(all(v));
ll by = -1, bx = -1, t = 0;
for(int i = 0; i < n; i++){
auto&[x, y] = v[i];
if(bx == x && y == by + 1) a[i] = a[i - 1];
else a[i] = ++t;
w[a[i]]++;
int j = lower_bound(all(v), make_pair(x - 1, y)) - v.begin();
if(v[j] == make_pair(x - 1, y)) {
adj[a[j]].emplace_back(a[i]);
adj[a[i]].emplace_back(a[j]);
}
by = y; bx = x;
}
for(int i = 1; i <= t; i++) adj[i].erase(unique(all(adj[i])), adj[i].end());
ll ret = dfs(1, -1);
for(int i = 1; i <= t; i++) adj[i].clear(), w[i] = 0;
return ret;
}
int DistanceSum(int n, int* X, int* Y){
ll ret = 0;
for(int i = 0; i < n; i++) v.emplace_back(X[i], Y[i]);
ret = go();
for(int i = 0; i < n; i++) swap(v[i].first, v[i].second);
ret += go(); ret %= mod;
return ret;
}
int main(void){
ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
cin >> n;
for(int i = 0; i < n; i++) cin >> X[i] >> Y[i];
cout << DistanceSum(n, X, Y);
return 0;
}
Compilation message
/usr/bin/ld: /tmp/ccVPgZg1.o: in function `main':
city.cpp:(.text.startup+0x0): multiple definition of `main'; /tmp/ccC6gOd5.o:grader.cpp:(.text.startup+0x0): first defined here
collect2: error: ld returned 1 exit status