oj.uz

답안 #519873

# 제출 시각 아이디 문제 언어 결과 실행 시간 메모리
519873 2022-01-27T13:27:01 Z erke The Xana coup (BOI21_xanadu) C++11
100 / 100
 78 ms 15324 KB 
#include <bits/stdc++.h>
using namespace std;

typedef long long ll;

const int N = 100005;
const int oo = 1e9;

vector<int> adj[N];
int color[N], dp[N][2][2];

template<typename T>
void chmin(T &a, T b) {
  if (a > b) a = b;
}

void dfs(int i, int p) {
  if (color[i] == 0) {
    dp[i][0][0] = 0;
    dp[i][1][1] = 1;
  }
  else if (color[i] == 1) {
    dp[i][1][0] = 0;
    dp[i][0][1] = 1;
  }
  for (auto &j : adj[i]) {
    if (j == p) continue;
    dfs(j, i);
    array<array<int,2>,2> cur_dp;
    cur_dp[0][0] = min(dp[i][0][0] + dp[j][0][0], dp[i][1][0] + dp[j][0][1]);
    cur_dp[0][1] = min(dp[i][0][1] + dp[j][1][0], dp[i][1][1] + dp[j][1][1]);
    cur_dp[1][0] = min(dp[i][0][0] + dp[j][0][1], dp[i][1][0] + dp[j][0][0]);
    cur_dp[1][1] = min(dp[i][0][1] + dp[j][1][1], dp[i][1][1] + dp[j][1][0]);
    for (int k = 0; k <= 1; k++)
    for (int l = 0; l <= 1; l++) {
      if (cur_dp[k][l] >= oo) cur_dp[k][l] = oo;
      dp[i][k][l] = cur_dp[k][l];
    }
  }
  /*cerr << "i = " << i << '\n';
  for (int k = 0; k <= 1; k++)
  for (int l = 0; l <= 1; l++) {
    if (dp[i][k][l] >= oo) {
      cerr << "-" << " \n"[l == 1];
    }
    else cerr << dp[i][k][l] << " \n"[l == 1];
  }*/
}

signed main() {
  cin.tie(0)->sync_with_stdio(0);
  int n; cin >> n;
  for (int i = 1; i < n; i++) {
    int u, v;
    cin >> u >> v;
    adj[u].push_back(v);
    adj[v].push_back(u);
  }
  for (int i = 1; i <= n; i++) {
    cin >> color[i];
  }
  for (int i = 0; i < N; i++)
  for (int j = 0; j < 2; j++)
  for (int k = 0; k < 2; k++) {
    dp[i][j][k] = oo;
  }
  dfs(1, -1);
  int ans = min(dp[1][0][0], dp[1][0][1]);
  if (ans >= oo) cout << "impossible" << '\n';
  else cout << ans << '\n';
}

Subtask #1
5.0 / 5.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 4172 KB Output is correct
2 Correct 2 ms 4172 KB Output is correct
3 Correct 2 ms 4208 KB Output is correct
4 Correct 2 ms 4208 KB Output is correct
5 Correct 2 ms 4172 KB Output is correct

Subtask #2
15.0 / 15.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 4172 KB Output is correct
2 Correct 2 ms 4172 KB Output is correct
3 Correct 2 ms 4208 KB Output is correct
4 Correct 2 ms 4208 KB Output is correct
5 Correct 2 ms 4172 KB Output is correct
6 Correct 2 ms 4172 KB Output is correct
7 Correct 2 ms 4204 KB Output is correct
8 Correct 2 ms 4172 KB Output is correct
9 Correct 3 ms 4172 KB Output is correct
10 Correct 3 ms 4172 KB Output is correct
11 Correct 2 ms 4172 KB Output is correct
12 Correct 3 ms 4172 KB Output is correct
13 Correct 3 ms 4212 KB Output is correct
14 Correct 2 ms 4172 KB Output is correct
15 Correct 3 ms 4172 KB Output is correct
16 Correct 2 ms 4172 KB Output is correct

Subtask #3
10.0 / 10.0

# 결과 실행 시간 메모리 Grader output
1 Correct 49 ms 15224 KB Output is correct
2 Correct 53 ms 15116 KB Output is correct
3 Correct 47 ms 15324 KB Output is correct

Subtask #4
40.0 / 40.0

# 결과 실행 시간 메모리 Grader output
1 Correct 50 ms 15160 KB Output is correct
2 Correct 49 ms 15044 KB Output is correct
3 Correct 49 ms 15300 KB Output is correct
4 Correct 40 ms 8476 KB Output is correct
5 Correct 48 ms 8956 KB Output is correct
6 Correct 78 ms 9032 KB Output is correct
7 Correct 3 ms 4212 KB Output is correct
8 Correct 16 ms 5792 KB Output is correct
9 Correct 44 ms 8976 KB Output is correct
10 Correct 55 ms 9076 KB Output is correct
11 Correct 58 ms 9412 KB Output is correct
12 Correct 59 ms 9632 KB Output is correct
13 Correct 45 ms 8736 KB Output is correct
14 Correct 61 ms 9208 KB Output is correct

Subtask #5
30.0 / 30.0

# 결과 실행 시간 메모리 Grader output
1 Correct 2 ms 4172 KB Output is correct
2 Correct 2 ms 4172 KB Output is correct
3 Correct 2 ms 4208 KB Output is correct
4 Correct 2 ms 4208 KB Output is correct
5 Correct 2 ms 4172 KB Output is correct
6 Correct 2 ms 4172 KB Output is correct
7 Correct 2 ms 4204 KB Output is correct
8 Correct 2 ms 4172 KB Output is correct
9 Correct 3 ms 4172 KB Output is correct
10 Correct 3 ms 4172 KB Output is correct
11 Correct 2 ms 4172 KB Output is correct
12 Correct 3 ms 4172 KB Output is correct
13 Correct 3 ms 4212 KB Output is correct
14 Correct 2 ms 4172 KB Output is correct
15 Correct 3 ms 4172 KB Output is correct
16 Correct 2 ms 4172 KB Output is correct
17 Correct 49 ms 15224 KB Output is correct
18 Correct 53 ms 15116 KB Output is correct
19 Correct 47 ms 15324 KB Output is correct
20 Correct 50 ms 15160 KB Output is correct
21 Correct 49 ms 15044 KB Output is correct
22 Correct 49 ms 15300 KB Output is correct
23 Correct 40 ms 8476 KB Output is correct
24 Correct 48 ms 8956 KB Output is correct
25 Correct 78 ms 9032 KB Output is correct
26 Correct 3 ms 4212 KB Output is correct
27 Correct 16 ms 5792 KB Output is correct
28 Correct 44 ms 8976 KB Output is correct
29 Correct 55 ms 9076 KB Output is correct
30 Correct 58 ms 9412 KB Output is correct
31 Correct 59 ms 9632 KB Output is correct
32 Correct 45 ms 8736 KB Output is correct
33 Correct 61 ms 9208 KB Output is correct
34 Correct 2 ms 4172 KB Output is correct
35 Correct 2 ms 4168 KB Output is correct
36 Correct 2 ms 4204 KB Output is correct
37 Correct 2 ms 4172 KB Output is correct
38 Correct 3 ms 4200 KB Output is correct
39 Correct 2 ms 4172 KB Output is correct
40 Correct 3 ms 4172 KB Output is correct
41 Correct 2 ms 4172 KB Output is correct
42 Correct 2 ms 4172 KB Output is correct
43 Correct 2 ms 4172 KB Output is correct
44 Correct 2 ms 4208 KB Output is correct
45 Correct 47 ms 15312 KB Output is correct
46 Correct 48 ms 15096 KB Output is correct
47 Correct 45 ms 15304 KB Output is correct
48 Correct 55 ms 8484 KB Output is correct
49 Correct 52 ms 8948 KB Output is correct
50 Correct 57 ms 9028 KB Output is correct
51 Correct 2 ms 4172 KB Output is correct
52 Correct 15 ms 5864 KB Output is correct
53 Correct 51 ms 8952 KB Output is correct
54 Correct 50 ms 9140 KB Output is correct
55 Correct 48 ms 9488 KB Output is correct
56 Correct 57 ms 9652 KB Output is correct
57 Correct 50 ms 8688 KB Output is correct
58 Correct 47 ms 9068 KB Output is correct
59 Correct 14 ms 5748 KB Output is correct
60 Correct 45 ms 8620 KB Output is correct
61 Correct 53 ms 8948 KB Output is correct
62 Correct 54 ms 9084 KB Output is correct
63 Correct 47 ms 9156 KB Output is correct
64 Correct 49 ms 9144 KB Output is correct
65 Correct 44 ms 9612 KB Output is correct
66 Correct 48 ms 9516 KB Output is correct
67 Correct 35 ms 9496 KB Output is correct
68 Correct 32 ms 9404 KB Output is correct
69 Correct 40 ms 9488 KB Output is correct
70 Correct 52 ms 9444 KB Output is correct