T2 给用户推荐股票的数目

每个用户都有喜欢的股票，如果一个用户 A 喜欢股票 a, b；用户 B 喜欢 b, c。那么相当于在股票 a 和 b, b 和 c 之间建立了两条边，如果用户 C 喜欢 c，那么系统会给 c 推荐股票 a 和 b。

求给用户 x 推荐股票的数目（不包含自己喜欢的股票）

#include <bits/stdc++.h>

using namespace std;

#define maxn 50005

struct UF {
    int f[maxn+5], cnt[maxn+5];

    UF(int n) {
        for (int i = 0; i <=n; ++i) {
            f[i] = i; cnt[i] = 1;
        }
    }
    int getp(int x) { return f[x] == x ? f[x] : f[x] = getp(f[x]); }
    void add(int x, int y) {
        int px = getp(x), py = getp(y);
        if (cnt[px] < cnt[py]) swap(px, py);
        if (px != py) {
            f[px] = py;
            cnt[py] += cnt[px];
        }
    }
};

int main() {
    int q; cin >> q;
    int n, op, no = 0;
    string name, share;
    unordered_map<string, set<int>> users;
    unordered_map<string, int> shareid;
    UF uf(maxn);
    while (q--) {
        cin >> op >> name;
        if (op == 1) {
            cin >> n;
            vector<int> ids(n);
            for (int i = 0; i < n; ++i) {
                cin >> share;
                if (!shareid.count(share)) shareid[share] = no++;
                ids[i] = shareid[share];
                users[name].insert(ids[i]);
            }
            for (int i = 0; i < n; ++i) {
                for (int j = i+1; j < n; ++j) {
                    uf.add(ids[i], ids[j]);
                }
            }
        } else {
            if (users[name].empty()) {
                cout << "error" << endl;
                continue;
            }
            int inc = 0;
            set<int> pp;
            for (auto us : users[name]) {
                int p = uf.getp(us);
                if (pp.count(p)) continue;
                pp.insert(p);
                auto cnt = uf.cnt[p];
                inc += cnt;
                for (auto us2 : users[name]) {
                    if (uf.getp(us) == uf.getp(us2)) --inc;
                }
            }
            cout << inc << endl;
        }
    }
    return 0;
}