QOJ.ac

QOJ

ID题目提交者结果用时内存语言文件大小提交时间测评时间
#180669#4891. 树上的孤独hos_lyric#0 13ms3968kbC++144.6kb2023-09-16 06:28:512024-07-04 02:00:47

Judging History

你现在查看的是最新测评结果

  • [2024-07-04 02:00:47]
  • 评测
  • 测评结果:0
  • 用时:13ms
  • 内存:3968kb
  • [2023-09-16 06:28:51]
  • 提交

answer

#include <cassert>
#include <cmath>
#include <cstdint>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
#include <bitset>
#include <complex>
#include <deque>
#include <functional>
#include <iostream>
#include <limits>
#include <map>
#include <numeric>
#include <queue>
#include <set>
#include <sstream>
#include <string>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>

using namespace std;

using Int = long long;

template <class T1, class T2> ostream &operator<<(ostream &os, const pair<T1, T2> &a) { return os << "(" << a.first << ", " << a.second << ")"; };
template <class T> ostream &operator<<(ostream &os, const vector<T> &as) { const int sz = as.size(); os << "["; for (int i = 0; i < sz; ++i) { if (i >= 256) { os << ", ..."; break; } if (i > 0) { os << ", "; } os << as[i]; } return os << "]"; }
template <class T> void pv(T a, T b) { for (T i = a; i != b; ++i) cerr << *i << " "; cerr << endl; }
template <class T> bool chmin(T &t, const T &f) { if (t > f) { t = f; return true; } return false; }
template <class T> bool chmax(T &t, const T &f) { if (t < f) { t = f; return true; } return false; }
#define COLOR(s) ("\x1b[" s "m")


int N[2], Q;
vector<int> A[2], B[2];
vector<int> C[2];
vector<int> O, X[4];
int K;

vector<vector<int>> graph[2];
int zeit[2];
vector<int> dep[2], dis[2], fin[2];
void dfs(int h, int u, int p) {
  dep[h][u] = (~p) ? (dep[h][p] + 1) : 0;
  dis[h][u] = zeit[h]++;
  for (const int v : graph[h][u]) if (p != v) {
    dfs(h, v, u);
  }
  fin[h][u] = zeit[h];
}

namespace brute {
vector<int> run() {
  vector<int> css[2] = {C[0], C[1]};
  int lt = 0;
  vector<int> ans(Q, 0);
  vector<int> on(K, -1);
  for (int q = 0; q < Q; ++q) {
    if (O[q] == 1) {
      const int R[2] = {X[0][q], X[1][q]};
      const int D[2] = {X[2][q] ^ lt, X[3][q] ^ lt};
// cerr<<COLOR("33")<<"calc "<<R[0]<<","<<D[0]<<" "<<R[1]<<","<<D[1]<<COLOR()<<endl;
      for (int h = 0; h < 2; ++h) for (int u = 0; u < N[h]; ++u) {
        if (dis[h][R[h]] <= dis[h][u] && dis[h][u] < fin[h][R[h]] && dep[h][u] <= dep[h][R[h]] + D[h]) {
// cerr<<"  "<<h<<" "<<u<<endl;
          on[css[h][u]] = q;
        }
      }
      for (int k = 0; k < K; ++k) if (on[k] == q) {
        ++ans[q];
      }
      lt = ans[q];
    } else if (O[q] == 2) {
// cerr<<COLOR("33")<<"change "<<X[0][q]<<" "<<X[1][q]<<endl;
      css[0][X[0][q]] = X[1][q];
    } else {
      assert(false);
    }
  }
  return ans;
}
}  // brute


int main() {
  for (; ~scanf("%d%d%d", &N[0], &N[1], &Q); ) {
    for (int h = 0; h < 2; ++h) {
      A[h].resize(N[h] - 1);
      B[h].resize(N[h] - 1);
      for (int i = 0; i < N[h] - 1; ++i) {
        scanf("%d%d", &A[h][i], &B[h][i]);
        --A[h][i];
        --B[h][i];
      }
    }
    for (int h = 0; h < 2; ++h) {
      C[h].resize(N[h]);
      for (int u = 0; u < N[h]; ++u) {
        scanf("%d", &C[h][u]);
      }
    }
    O.resize(Q);
    X[0].assign(Q, 0);
    X[1].assign(Q, 0);
    X[2].assign(Q, 0);
    X[3].assign(Q, 0);
    for (int q = 0; q < Q; ++q) {
      scanf("%d%d%d", &O[q], &X[0][q], &X[1][q]);
      if (O[q] == 1) {
        scanf("%d%d", &X[2][q], &X[3][q]);
        --X[0][q];
        --X[1][q];
      } else if (O[q] == 2) {
        --X[0][q];
      } else {
        assert(false);
      }
    }
    
    {
      vector<int> cs;
      for (int h = 0; h < 2; ++h) for (int u = 0; u < N[h]; ++u) {
        cs.push_back(C[h][u]);
      }
      for (int q = 0; q < Q; ++q) if (O[q] == 2) {
        cs.push_back(X[1][q]);
      }
      sort(cs.begin(), cs.end());
      cs.erase(unique(cs.begin(), cs.end()), cs.end());
      K = cs.size();
      auto indexOf = [&](int c) -> int {
        return lower_bound(cs.begin(), cs.end(), c) - cs.begin();
      };
      for (int h = 0; h < 2; ++h) for (int u = 0; u < N[h]; ++u) {
        C[h][u] = indexOf(C[h][u]);
      }
      for (int q = 0; q < Q; ++q) if (O[q] == 2) {
        cs[q] = indexOf(cs[q]);
      }
    }
// cerr<<"K = "<<K<<", C[0] = "<<C[0]<<", C[1] = "<<C[1]<<endl;
    
    for (int h = 0; h < 2; ++h) {
      graph[h].assign(N[h], {});
      for (int i = 0; i < N[h] - 1; ++i) {
        graph[h][A[h][i]].push_back(B[h][i]);
        graph[h][B[h][i]].push_back(A[h][i]);
      }
      zeit[h] = 0;
      dep[h].assign(N[h], -1);
      dis[h].assign(N[h], -1);
      fin[h].assign(N[h], -1);
      dfs(h, 0, -1);
    }
    
    const auto ans = brute::run();
    for (int q = 0; q < Q; ++q) if (O[q] == 1) {
      printf("%d\n", ans[q]);
    }
  }
  return 0;
}

詳細信息

Subtask #1:

score: 0
Wrong Answer

Test #1:

score: 0
Wrong Answer
time: 13ms
memory: 3968kb

input:

20 2000 2000
8 15
8 13
14 8
14 7
12 14
12 1
9 13
9 4
5 9
5 10
2 5
2 19
6 19
6 16
11 7
11 20
18 2
18 17
3 6
1395 1783
1395 1735
1735 457
457 739
739 438
438 101
101 441
441 1879
1879 1238
1238 501
501 1732
1732 1910
1910 2000
2000 834
834 917
917 111
111 780
780 1966
1966 1604
1604 623
623 1748
1748 ...

output:

105
92
2
44
18
53
193
24
69
20
21
200
78
87
136
31
68
68
52
36
24
10
24
78
59
100
27
200
39
79
97
21
88
35
197
77
28
49
11
20
54
60
24
5
110
158
24
22
15
105
46
108
27
75
169
83
31
146
144
9
136
33
110
140
121
68
33
68
19
43
57
50
57
14
36
145
113
137
145
200
22
81
82
38
73
27
30
38
161
51
61
53
95
...

result:

wrong answer 2nd numbers differ - expected: '93', found: '92'

Subtask #2:

score: 0
Time Limit Exceeded

Test #3:

score: 0
Time Limit Exceeded

input:

20 200000 500000
8 18
8 4
2 4
2 14
13 4
13 16
6 4
6 3
1 4
1 17
15 6
15 19
7 17
7 11
5 14
5 10
20 7
12 15
9 8
165302 77239
165302 198189
165302 180850
165302 192738
165302 173589
165302 194087
165302 191990
165302 167370
165302 101092
165302 92553
165302 163654
165302 122381
165302 152105
165302 1919...

output:


result:


Subtask #3:

score: 0
Time Limit Exceeded

Test #7:

score: 0
Time Limit Exceeded

input:

20 100000 100000
16 12
16 20
6 12
6 18
2 16
2 8
5 20
5 13
3 6
3 11
19 11
19 17
9 12
9 15
4 15
4 7
10 5
14 15
14 1
85812 94626
85812 91172
91172 93788
93788 96567
96567 75524
75524 23275
23275 98340
98340 81608
81608 91480
91480 75108
75108 56605
56605 93317
93317 41617
41617 77160
77160 727
727 7559...

output:


result:


Subtask #4:

score: 0
Time Limit Exceeded

Test #13:

score: 0
Time Limit Exceeded

input:

1 200000 500000
189127 137023
189127 199761
199761 160701
160701 130639
130639 190908
190908 176819
176819 193363
193363 188021
188021 182446
182446 186028
186028 198828
198828 190792
190792 155378
155378 189428
189428 177276
177276 146159
146159 175923
175923 188073
188073 182206
182206 131072
1310...

output:


result:


Subtask #5:

score: 0
Runtime Error

Test #15:

score: 0
Runtime Error

input:

20 200000 1000000
13 10
13 5
19 5
19 20
15 10
15 6
12 6
12 3
8 10
8 2
1 20
1 11
14 10
14 16
18 3
18 7
4 3
9 10
9 17
194055 154514
194055 148156
148156 115271
115271 198116
198116 179433
179433 171975
171975 196600
196600 167194
167194 185078
185078 191409
191409 163496
163496 178243
178243 154093
15...

output:


result: