比赛 寒假集训2 评测结果 AAATATEEEEEEEEEEEEEE
题目名称 轻重边 最终得分 20
用户昵称 dbk 运行时间 4.788 s
代码语言 C++ 内存使用 4.01 MiB
提交时间 2026-02-25 11:45:36
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
const int N = 10100; 
vector<int> g[N];
int f[N], d[N];
unordered_map<int, unordered_map<int, int>> e;
void dfs(int u, int fa) {
    f[u] = fa;
    d[u] = d[fa] + 1;
    for (int v : g[u]) {
        if (v != fa) {
            dfs(v, u);
        }
    }
}
void get_pn(int a, int b, vector<int>& pn) {
    pn.clear();
    vector<int> pa, pb;
    while (a != b) {
        if (d[a] > d[b]) {
            pa.push_back(a);
            a = f[a];
        } else {
            pb.push_back(b);
            b = f[b];
        }
    }
    pa.push_back(a);
    reverse(pb.begin(), pb.end());
    pn = pa;
    for (int x : pb) pn.push_back(x);
}
void get_pe(int a, int b, vector<pair<int, int>>& pe) {
    pe.clear();
    while (a != b) {
        if (d[a] > d[b]) {
            pe.push_back({a, f[a]});
            a = f[a];
        } else {
            pe.push_back({b, f[b]});
            b = f[b];
        }
    }
}
void clr_e(const vector<int>& pn) {
    for (int x : pn) {
        for (int v : g[x]) {
            e[x][v] = 0;
            e[v][x] = 0;
        }
    }
}
void set_h(const vector<pair<int, int>>& pe) {
    for (auto& ed : pe) {
        int u = ed.first, v = ed.second;
        e[u][v] = 1;
        e[v][u] = 1;
    }
}
int cnt_h(const vector<pair<int, int>>& pe) {
    int cnt = 0;
    for (auto& ed : pe) {
        int u = ed.first, v = ed.second;
        if (e[u][v] == 1) cnt++;
    }
    return cnt;
}
void rst(int n) {
    for (int i = 1; i <= n; i++) {
        g[i].clear();
        f[i] = d[i] = 0;
    }
    e.clear();
}
int main() {
    freopen("edge.in", "r", stdin);
    freopen("edge.out", "w", stdout); 
    ios::sync_with_stdio(false);
    cin.tie(0);
    int T;
    cin >> T;
    while (T--) {
        int n, m;
        cin >> n >> m;
        rst(n);
        for (int i = 1; i < n; i++) {
            int u, v;
            cin >> u >> v;
            g[u].push_back(v);
            g[v].push_back(u);
            e[u][v] = 0;
            e[v][u] = 0;
        }
        d[0] = 0;
        dfs(1, 0);
        while (m--) {
            int op, a, b;
            cin >> op >> a >> b;
            if (op == 1) {
                vector<int> pn;
                get_pn(a, b, pn);
                clr_e(pn);
                
                vector<pair<int, int>> pe;
                get_pe(a, b, pe);
                set_h(pe);
            } else if (op == 2) {
                vector<pair<int, int>> pe;
                get_pe(a, b, pe);
                cout << cnt_h(pe) << '\n';
            }
        }
    }
    return 0;
}