Submission #967125

#TimeUsernameProblemLanguageResultExecution timeMemory
967125sleepntsheepCapital City (JOI20_capital_city)C11
100 / 100
371 ms16304 KiB
#pragma GCC optimize("O3, unroll-loops") #include <stdio.h> #define N 200000 #define M (2*N-2) #define N_ N+1 #define M_ M+1 #define K N int n, k, color[N], answer; /* graph */ int head[N], nxt[M_], vv[M_]; /* node with color */ int head_[K], nxt_[N_], vv_[N_]; void link(int*head, int*nxt, int*vv, int*i, int u, int v) { nxt[i[0]] = head[u]; vv[i[0]] = v; head[u] = i[0]++; } int dead[N], sz[N]; void dfs2(int u, int p) { sz[u] = 1; for (int j = head[u]; j; j = nxt[j]) { if (vv[j] == p || dead[vv[j]]) continue; dfs2(vv[j], u); sz[u] += sz[vv[j]]; } } int par[N], visit[K], dfreq[K], cfreq[K]; void dfs3(int u, int p) { visit[color[u]] = -1; ++dfreq[color[u]]; par[u] = p; for (int j = head[u]; j; j = nxt[j]) { if (vv[j] == p || dead[vv[j]]) continue; dfs3(vv[j], u); } } void dfs4(int u, int p) { visit[color[u]] = -1; dfreq[color[u]] = 0; for (int j = head[u]; j; j = nxt[j]) { if (vv[j] == p || dead[vv[j]]) continue; dfs4(vv[j], u); } } int get_centroid(int u, int p, int tsize) { for (int j = head[u]; j; j = nxt[j]) if (!dead[vv[j]] && vv[j] != p && sz[vv[j]] * 2 >= tsize) return get_centroid(vv[j], u, tsize); return u; } void centroid_decomposition(int u) { dfs2(u, u); u = get_centroid(u, u, sz[u]); dead[u] = 1; dfs3(u, u); int col = color[u]; if (dfreq[col] == cfreq[col]) { static int qu[N], head, tail; head = tail = 0; int dcolor = 0; for (int at, jj = head_[col]; jj; jj = nxt_[jj]) qu[head++] = vv_[jj]; visit[col] = 1; while (head-tail) { int uu = qu[tail++]; if (uu == u || visit[color[par[uu]]] == 1) continue; int nextcol = color[par[uu]]; if (cfreq[nextcol] != dfreq[nextcol]) goto NOTOK; ++dcolor; for (int jj = head_[nextcol]; jj; jj = nxt_[jj]) qu[head++] = vv_[jj]; visit[nextcol] = 1; } if (dcolor < answer) answer = dcolor; NOTOK:; } dfs4(u, u); for (int j = head[u]; j; j = nxt[j]) if (!dead[vv[j]]) centroid_decomposition(vv[j]); } int main() { scanf("%d%d", &n, &k); answer = n; for (int i = 1, u, v, eo[1] = { 1 }; i < n; ++i) { scanf("%d%d", &u, &v); --u, --v; link(head, nxt, vv, eo, u, v); link(head, nxt, vv, eo, v, u); } for (int eo[1] = { 1 }, i = 0; i < n; ++i) { scanf("%d", color+i); ++cfreq[--color[i]]; link(head_, nxt_, vv_, eo, color[i], i); } centroid_decomposition(0); printf("%d", answer); }

Compilation message (stderr)

capital_city.c:1:9: warning: bad option '-f unroll-loops' to pragma 'optimize' [-Wpragmas]
    1 | #pragma GCC optimize("O3, unroll-loops")
      |         ^~~
capital_city.c:19:1: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   19 | {
      | ^
capital_city.c:28:1: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   28 | {
      | ^
capital_city.c:42:1: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   42 | {
      | ^
capital_city.c:55:1: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   55 | {
      | ^
capital_city.c:67:1: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   67 | {
      | ^
capital_city.c:75:1: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
   75 | {
      | ^
capital_city.c: In function 'centroid_decomposition':
capital_city.c:89:18: warning: unused variable 'at' [-Wunused-variable]
   89 |         for (int at, jj = head_[col]; jj; jj = nxt_[jj])
      |                  ^~
capital_city.c: At top level:
capital_city.c:119:1: warning: bad option '-f unroll-loops' to attribute 'optimize' [-Wattributes]
  119 | {
      | ^
capital_city.c: In function 'main':
capital_city.c:120:5: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
  120 |     scanf("%d%d", &n, &k);
      |     ^~~~~~~~~~~~~~~~~~~~~
capital_city.c:124:9: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
  124 |         scanf("%d%d", &u, &v); --u, --v;
      |         ^~~~~~~~~~~~~~~~~~~~~
capital_city.c:130:9: warning: ignoring return value of 'scanf' declared with attribute 'warn_unused_result' [-Wunused-result]
  130 |         scanf("%d", color+i);
      |         ^~~~~~~~~~~~~~~~~~~~
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...