#pragma GCC optimize("Ofast") #include using namespace std; using lint = long long; using pint = pair; using plint = pair; struct fast_ios { fast_ios(){ cin.tie(nullptr); ios::sync_with_stdio(false); cout << fixed << setprecision(20); }; } fast_ios_; #define ALL(x) (x).begin(), (x).end() #define FOR(i, begin, end) for(int i=(begin),i##_end_=(end);i=i##_begin_;i--) #define REP(i, n) FOR(i,0,n) #define IREP(i, n) IFOR(i,0,n) template void ndarray(vector &vec, int len) { vec.resize(len); } template void ndarray(vector &vec, int len, Args... args) { vec.resize(len); for (auto &v : vec) ndarray(v, args...); } template void ndfill(V &x, const T &val) { x = val; } template void ndfill(vector &vec, const T &val) { for (auto &v : vec) ndfill(v, val); } template bool chmax(T &m, const T q) { if (m < q) {m = q; return true;} else return false; } template bool chmin(T &m, const T q) { if (m > q) {m = q; return true;} else return false; } template pair operator+(const pair &l, const pair &r) { return make_pair(l.first + r.first, l.second + r.second); } template pair operator-(const pair &l, const pair &r) { return make_pair(l.first - r.first, l.second - r.second); } template vector srtunq(vector vec) { sort(vec.begin(), vec.end()), vec.erase(unique(vec.begin(), vec.end()), vec.end()); return vec; } template istream &operator>>(istream &is, vector &vec) { for (auto &v : vec) is >> v; return is; } template ostream &operator<<(ostream &os, const vector &vec) { os << '['; for (auto v : vec) os << v << ','; os << ']'; return os; } #if __cplusplus >= 201703L template istream &operator>>(istream &is, tuple &tpl) { std::apply([&is](auto &&... args) { ((is >> args), ...);}, tpl); return is; } template ostream &operator<<(ostream &os, const tuple &tpl) { std::apply([&os](auto &&... args) { ((os << args << ','), ...);}, tpl); return os; } #endif template ostream &operator<<(ostream &os, const deque &vec) { os << "deq["; for (auto v : vec) os << v << ','; os << ']'; return os; } template ostream &operator<<(ostream &os, const set &vec) { os << '{'; for (auto v : vec) os << v << ','; os << '}'; return os; } template ostream &operator<<(ostream &os, const unordered_set &vec) { os << '{'; for (auto v : vec) os << v << ','; os << '}'; return os; } template ostream &operator<<(ostream &os, const multiset &vec) { os << '{'; for (auto v : vec) os << v << ','; os << '}'; return os; } template ostream &operator<<(ostream &os, const unordered_multiset &vec) { os << '{'; for (auto v : vec) os << v << ','; os << '}'; return os; } template ostream &operator<<(ostream &os, const pair &pa) { os << '(' << pa.first << ',' << pa.second << ')'; return os; } template ostream &operator<<(ostream &os, const map &mp) { os << '{'; for (auto v : mp) os << v.first << "=>" << v.second << ','; os << '}'; return os; } template ostream &operator<<(ostream &os, const unordered_map &mp) { os << '{'; for (auto v : mp) os << v.first << "=>" << v.second << ','; os << '}'; return os; } #ifdef HITONANODE_LOCAL #define dbg(x) cerr << #x << " = " << (x) << " (L" << __LINE__ << ") " << __FILE__ << endl #else #define dbg(x) #endif vector depth, depthinv; vector vs, vsinv; vector> to; void dfs_lca(int now, int prv, int d) { depth[now] = d; vs[now] = vsinv.size(); vsinv.emplace_back(now); depthinv.emplace_back(d); for (auto nxt : to[now]) if (nxt != prv) { dfs_lca(nxt, now, d + 1); vsinv.emplace_back(now); depthinv.emplace_back(d); } } struct rand_int_ { using lint = long long; mt19937 mt; rand_int_() : mt(chrono::steady_clock::now().time_since_epoch().count()) {} lint operator()(lint x) { return this->operator()(0, x); } // [0, x) lint operator()(lint l, lint r) { uniform_int_distribution d(l, r - 1); return d(mt); } } rnd; // Static sequence sparse table // Complexity: O(NlogN) for precalculation, O(1) per query template struct SparseTable { int N, lgN; T defaultT; std::vector> data; std::vector lgx_table; SparseTable(const std::vector &sequence, T defaultT) : N(sequence.size()), defaultT(defaultT) { lgx_table.resize(N + 1); for (int i = 2; i < N + 1; i++) lgx_table[i] = lgx_table[i >> 1] + 1; lgN = lgx_table[N] + 1; data.assign(lgN, std::vector(N, defaultT)); data[0] = sequence; for (int d = 1; d < lgN; d++) { for (int i = 0; i + (1 << d) <= N; i++) { data[d][i] = min(data[d - 1][i], data[d - 1][i + (1 << (d - 1))]); } } } T get(int l, int r) { // [l, r), 0-indexed assert(l >= 0 and r <= N); int d = lgx_table[r - l]; return min(data[d][l], data[d][r - (1 << d)]); } }; int main() { int N, K, Q; cin >> N >> K >> Q; int root = rnd(N); vector C(K); cin >> C; for (auto &a : C) a--; vector cnt(N); for (auto a : C) cnt[a]++; to.resize(N); REP(_, N - 1) { int a, b; cin >> a >> b; a--, b--; to[a].emplace_back(b); to[b].emplace_back(a); } depth.assign(N, 0); depthinv.clear(); vs.assign(N, -1); vsinv.clear(); dfs_lca(root, -1, 0); dbg(depth); dbg(depthinv); dbg(vsinv); SparseTable rmq(depthinv, 1e9); auto distance = [&](int i, int j) -> lint { if (i == j) return 0; i = vs[i]; j = vs[j]; if (i > j) swap(i, j); return depthinv[i] + depthinv[j] - rmq.get(i, j + 1) * 2; }; vector tot(N), totch(N); vector ch(N); auto dfs2 = [&](auto &&dfs2, int now, int prv) -> void { for (auto nxt : to[now]) if (nxt != prv) { dfs2(dfs2, nxt, now); ch[now] += ch[nxt]; totch[now] += totch[nxt] + ch[nxt]; } }; auto dfs3 = [&](auto &&dfs3, int now, int prv) -> void { tot[now] = totch[now]; if (prv != -1) { tot[now] = tot[prv] - ch[now] + (K - ch[now]); } for (auto nxt : to[now]) if (nxt != prv) { dfs3(dfs3, nxt, now); } }; vector upd_info; auto refresh = [&]() { totch.assign(N, 0); tot.assign(N, 0); ch = cnt; dfs2(dfs2, root, -1); dfs3(dfs3, root, -1); }; refresh(); dbg(totch); dbg(tot); int nb_q = 0; REP(t, Q) { int q; cin >> q; if (q == 1) { int p, d; cin >> p >> d; p--, d--; cnt[C[p]]--; upd_info.emplace_back(C[p], -1); C[p] = d; cnt[C[p]]++; upd_info.emplace_back(C[p], 1); } if (upd_info.size() > 300 and nb_q > 100000) { refresh(); nb_q = 0; } if (q == 2) { int e; cin >> e; e--; lint ans = tot[e]; for (auto [i, w] : upd_info) { ans += w * distance(i, e); nb_q++; } cout << ans << '\n'; } } }