oj.uz

Submission #262611

# Submission time Handle Problem Language Result Execution time Memory
262611 2020-08-13T05:18:41 Z dennisstar Capital City (JOI20_capital_city) C++17
11 / 100
 1562 ms 524292 KB 
#include <bits/stdc++.h>
#define eb emplace_back
using namespace std;
using pii=pair<int, int>;

const int MX = 2e5 + 5;

int N, K, C[MX];
int A, vis[MX], sz[MX];
vector<int> adj[MX], ele[MX];

int P[MX], cnt[MX], use[MX], clh[MX];
void d1(int n, int p) {
	sz[n]=1; cnt[C[n]]++;
	for (auto &i:adj[n])
		if (i!=p&&!vis[i]) d1(i, n), sz[n]+=sz[i];
}

int d2(int n, int p, int c) {
	for (auto &i:adj[n])
		if (i!=p&&!vis[i]&&sz[i]>=c) return d2(i, n, c);
	return n;
}

void d3(int n) {
	vector<pii> V;
	V.eb(n, 0);
	for (int i=0; i<V.size(); i++) {
		int x, y;
		x=V[i].first, y=V[i].second;
		P[x]=y;
		for (auto &j:adj[x])
			if (j!=y&&!vis[j]) V.eb(j, x);
	}
}

void d4(int n) {
	vector<pii> V;
	V.eb(n, 0);
	for (int i=0; i<V.size(); i++) {
		int x, y;
		x=V[i].first, y=V[i].second;
		cnt[C[x]]=0;
		for (auto &j:adj[x])
			if (j!=y&&!vis[j]) V.eb(j, x);
	}
}

void sol(int st) {
	d1(st, 0);
	int c=d2(st, 0, sz[st]/2+1);
	d3(c);

	int fl=0, d=0;
	vector<int> s; s.eb(c);
	for (int i=0; i<s.size(); i++) {
		int x=s[i];
		if (use[x]) continue;
		if (ele[C[x]].size()!=cnt[C[x]]) { fl=1; break; }
		if (!clh[C[x]]) {
			clh[C[x]]=1;
			d++;
			for (auto &j:ele[C[x]]) s.eb(j);
		}
		if (P[x]) s.eb(P[x]);
	}
	if (!fl) A=min(A, d);
	for (auto &i:s) use[i]=0, clh[C[i]]=0;
	d4(st);
	s.clear();

	vis[c]=1;
	for (auto &i:adj[c]) if (!vis[i]) sol(i);
}

int main() {
	cin.tie(0)->sync_with_stdio(0);

	cin>>N>>K;
	for (int i=1, u, v; i<N; i++) cin>>u>>v, adj[u].eb(v), adj[v].eb(u);
	for (int i=1; i<=N; i++) cin>>C[i], ele[C[i]].eb(i);

	A=MX;
	sol(1);

	cout<<A-1<<'\n';
	return 0;
}

Compilation message

capital_city.cpp: In function 'void d3(int)':
capital_city.cpp:28:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   28 |  for (int i=0; i<V.size(); i++) {
      |                ~^~~~~~~~~
capital_city.cpp: In function 'void d4(int)':
capital_city.cpp:40:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   40 |  for (int i=0; i<V.size(); i++) {
      |                ~^~~~~~~~~
capital_city.cpp: In function 'void sol(int)':
capital_city.cpp:56:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   56 |  for (int i=0; i<s.size(); i++) {
      |                ~^~~~~~~~~
capital_city.cpp:59:23: warning: comparison of integer expressions of different signedness: 'std::vector<int>::size_type' {aka 'long unsigned int'} and 'int' [-Wsign-compare]
   59 |   if (ele[C[x]].size()!=cnt[C[x]]) { fl=1; break; }
      |       ~~~~~~~~~~~~~~~~^~~~~~~~~~~

Subtask #1
1.0 / 1.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 9728 KB Output is correct
2 Correct 8 ms 9728 KB Output is correct
3 Correct 7 ms 9728 KB Output is correct
4 Correct 7 ms 9728 KB Output is correct
5 Correct 8 ms 9728 KB Output is correct
6 Correct 7 ms 9728 KB Output is correct
7 Correct 7 ms 9728 KB Output is correct
8 Correct 7 ms 9728 KB Output is correct
9 Correct 7 ms 9728 KB Output is correct
10 Correct 7 ms 9728 KB Output is correct

Subtask #2
10.0 / 10.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 9728 KB Output is correct
2 Correct 8 ms 9728 KB Output is correct
3 Correct 7 ms 9728 KB Output is correct
4 Correct 7 ms 9728 KB Output is correct
5 Correct 8 ms 9728 KB Output is correct
6 Correct 7 ms 9728 KB Output is correct
7 Correct 7 ms 9728 KB Output is correct
8 Correct 7 ms 9728 KB Output is correct
9 Correct 7 ms 9728 KB Output is correct
10 Correct 7 ms 9728 KB Output is correct
11 Correct 11 ms 9984 KB Output is correct
12 Correct 10 ms 10112 KB Output is correct
13 Correct 10 ms 10112 KB Output is correct
14 Correct 10 ms 10112 KB Output is correct
15 Correct 10 ms 9984 KB Output is correct
16 Correct 11 ms 10112 KB Output is correct
17 Correct 15 ms 12148 KB Output is correct
18 Correct 15 ms 12148 KB Output is correct
19 Correct 16 ms 12148 KB Output is correct
20 Correct 15 ms 12148 KB Output is correct
21 Correct 15 ms 12148 KB Output is correct
22 Correct 9 ms 9984 KB Output is correct
23 Correct 9 ms 9984 KB Output is correct
24 Correct 13 ms 10612 KB Output is correct
25 Correct 25 ms 13800 KB Output is correct
26 Correct 24 ms 13936 KB Output is correct
27 Correct 26 ms 13800 KB Output is correct
28 Correct 18 ms 12148 KB Output is correct
29 Correct 18 ms 12148 KB Output is correct
30 Correct 18 ms 12656 KB Output is correct

Subtask #3
0 / 30.0

# Verdict Execution time Memory Grader output
1 Correct 767 ms 34800 KB Output is correct
2 Correct 319 ms 34964 KB Output is correct
3 Correct 758 ms 34712 KB Output is correct
4 Correct 311 ms 34972 KB Output is correct
5 Runtime error 1562 ms 524292 KB Execution killed with signal 9 (could be triggered by violating memory limits)
6 Halted 0 ms 0 KB -

Subtask #4
0 / 59.0

# Verdict Execution time Memory Grader output
1 Correct 7 ms 9728 KB Output is correct
2 Correct 8 ms 9728 KB Output is correct
3 Correct 7 ms 9728 KB Output is correct
4 Correct 7 ms 9728 KB Output is correct
5 Correct 8 ms 9728 KB Output is correct
6 Correct 7 ms 9728 KB Output is correct
7 Correct 7 ms 9728 KB Output is correct
8 Correct 7 ms 9728 KB Output is correct
9 Correct 7 ms 9728 KB Output is correct
10 Correct 7 ms 9728 KB Output is correct
11 Correct 11 ms 9984 KB Output is correct
12 Correct 10 ms 10112 KB Output is correct
13 Correct 10 ms 10112 KB Output is correct
14 Correct 10 ms 10112 KB Output is correct
15 Correct 10 ms 9984 KB Output is correct
16 Correct 11 ms 10112 KB Output is correct
17 Correct 15 ms 12148 KB Output is correct
18 Correct 15 ms 12148 KB Output is correct
19 Correct 16 ms 12148 KB Output is correct
20 Correct 15 ms 12148 KB Output is correct
21 Correct 15 ms 12148 KB Output is correct
22 Correct 9 ms 9984 KB Output is correct
23 Correct 9 ms 9984 KB Output is correct
24 Correct 13 ms 10612 KB Output is correct
25 Correct 25 ms 13800 KB Output is correct
26 Correct 24 ms 13936 KB Output is correct
27 Correct 26 ms 13800 KB Output is correct
28 Correct 18 ms 12148 KB Output is correct
29 Correct 18 ms 12148 KB Output is correct
30 Correct 18 ms 12656 KB Output is correct
31 Correct 767 ms 34800 KB Output is correct
32 Correct 319 ms 34964 KB Output is correct
33 Correct 758 ms 34712 KB Output is correct
34 Correct 311 ms 34972 KB Output is correct
35 Runtime error 1562 ms 524292 KB Execution killed with signal 9 (could be triggered by violating memory limits)
36 Halted 0 ms 0 KB -