oj.uz

Submission #62674

# Submission time Handle Problem Language Result Execution time Memory
62674 2018-07-29T18:18:06 Z zadrga Ideal city (IOI12_city) C++14
23 / 100
 107 ms 23216 KB 
// 19.28

#include <bits/stdc++.h>

using namespace std;

#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define INF ((int) 1e9)
#define MOD (1000 * 1000 * 1000)
#define maxn 100111

typedef long long ll;
typedef long double ld;
typedef pair<int, int> pii;

struct node{
	int ind, l, d, num;
	vector<int> adj;

	node(){
		ind = l = d = -1;
	}
} nodes[maxn + 2];

int n;
ll ans;

int sz[maxn];
vector<int> here[maxn + 2], cur[maxn + 2];

void DFS(int x, int p){
	sz[x] = nodes[x].num;
	for(int v : nodes[x].adj){
		if(v == p)
			continue;

		DFS(v, x);
		sz[x] += sz[v]; 
	}

	for(int v : nodes[x].adj){
		if(v == p)
			continue;

		ans = (ans + 1LL * sz[v] * (n - sz[v])) % MOD; 
	}
}

bool intersect(int x, int y){
	int zac = max(nodes[x].l, nodes[y].l);
	int kon = min(nodes[x].d, nodes[y].d);
	return (zac <= kon);
}

void solve(vector<pii> &v){
	for(int i = 0; i < maxn; i++){
		nodes[i].adj.clear();
		here[i].clear();
		cur[i].clear();
	}

	int cnt = 0;
	sort(v.begin(), v.end());
	for(int i = 0; i < v.size(); i++)
		here[v[i].fi].pb(v[i].se);

	for(int i = 0; i < maxn; i++){
		for(int j = 0; j < here[i].size(); j++){
			int k = j;
			while(k < here[i].size() && (here[i][k] - here[i][j] == k - j))
				k++;

			k--;
			nodes[++cnt].l = here[i][j];
			nodes[cnt].d = here[i][k];
			nodes[cnt].num = k - j + 1;
			cur[i].pb(cnt);

			j = k;
		}
	}

	for(int i = 0; i < maxn; i++){
		int ind = 0;
//		cout << i << ": " << endl;
		for(int j = 0; j < cur[i].size(); j++){
			int tren = cur[i][j];
//			cout << nodes[tren].l << "  " << nodes[tren].d << endl;

			while(ind < cur[i + 1].size() && nodes[cur[i + 1][ind]].d < nodes[cur[i][j]].l)
				ind++;

			while(ind < cur[i + 1].size() && intersect(cur[i][j], cur[i + 1][ind])){
				int x = cur[i][j];
				int y = cur[i + 1][ind];
				nodes[x].adj.pb(y);
				nodes[y].adj.pb(x);
				ind++;
			}
			ind--;
		}
//		cout << endl;
	}

	DFS(1, -1);
}

vector<int> mx, my;
vector<pii> v;

int DistanceSum(int N, int *X, int *Y) {
	n = N;
	for(int i = 0; i < n; i++){
		mx.pb(X[i]);
		my.pb(Y[i]);
	}

	sort(mx.begin(), mx.end());
	mx.resize(distance(mx.begin(), unique(mx.begin(), mx.end())));
	sort(my.begin(), my.end());
	my.resize(distance(my.begin(), unique(my.begin(), my.end())));

	for(int i = 0; i < n; i++){
		int x = lower_bound(mx.begin(), mx.end(), X[i]) - mx.begin();
		int y = lower_bound(my.begin(), my.end(), Y[i]) - my.begin();	
		v.pb(mp(x, y));
	}

	ans = 0;
	solve(v);

	for(int i = 0; i < n; i++)
		swap(v[i].fi, v[i].se);

	solve(v);

	return (int) ans;
}

Compilation message

city.cpp: In function 'void solve(std::vector<std::pair<int, int> >&)':
city.cpp:67:19: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i = 0; i < v.size(); i++)
                 ~~^~~~~~~~~~
city.cpp:71:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int j = 0; j < here[i].size(); j++){
                  ~~^~~~~~~~~~~~~~~~
city.cpp:73:12: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    while(k < here[i].size() && (here[i][k] - here[i][j] == k - j))
          ~~^~~~~~~~~~~~~~~~
city.cpp:89:20: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   for(int j = 0; j < cur[i].size(); j++){
                  ~~^~~~~~~~~~~~~~~
city.cpp:93:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    while(ind < cur[i + 1].size() && nodes[cur[i + 1][ind]].d < nodes[cur[i][j]].l)
          ~~~~^~~~~~~~~~~~~~~~~~~
city.cpp:96:14: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
    while(ind < cur[i + 1].size() && intersect(cur[i][j], cur[i + 1][ind])){
          ~~~~^~~~~~~~~~~~~~~~~~~
city.cpp:90:8: warning: unused variable 'tren' [-Wunused-variable]
    int tren = cur[i][j];
        ^~~~

Subtask #1
0 / 11.0

# Verdict Execution time Memory Grader output
1 Correct 13 ms 8952 KB Output is correct
2 Incorrect 13 ms 9036 KB Output isn't correct
3 Halted 0 ms 0 KB -

Subtask #2
0 / 21.0

# Verdict Execution time Memory Grader output
1 Incorrect 16 ms 9124 KB Output isn't correct
2 Halted 0 ms 0 KB -

Subtask #3
23.0 / 23.0

# Verdict Execution time Memory Grader output
1 Correct 34 ms 9876 KB Output is correct
2 Correct 39 ms 10384 KB Output is correct
3 Correct 65 ms 11388 KB Output is correct
4 Correct 54 ms 12108 KB Output is correct
5 Correct 103 ms 14124 KB Output is correct
6 Correct 100 ms 15504 KB Output is correct
7 Correct 102 ms 16260 KB Output is correct
8 Correct 87 ms 16376 KB Output is correct
9 Correct 88 ms 17864 KB Output is correct
10 Correct 107 ms 23216 KB Output is correct

Subtask #4
0 / 45.0

# Verdict Execution time Memory Grader output
1 Incorrect 32 ms 23216 KB Output isn't correct
2 Halted 0 ms 0 KB -