1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106
| #include <bits/stdc++.h> #define int long long const int MOD = 1e9 + 7;
struct DSU { std::vector<int> f, siz; DSU(int n) { f.resize(n); std::iota(f.begin(), f.end(), 0); siz.assign(n, 1); } int find(int x) { while (x != f[x]) { x = f[x] = f[f[x]]; } return x; } bool same(int x, int y) { return find(x) == find(y); }
bool merge(int x, int y) { x = find(x); y = find(y); if (x == y) { return false; } siz[x] += siz[y]; f[y] = x; return true; } int size(int x) { return siz[find(x)]; } }; void solve() { int n, m, k; std::cin >> n >> m >> k; std::vector<std::array<int, 3>> e(m);
for (int i = 0; i < m; i++) { int u, v, w; std::cin >> u >> v >> w; u--, v--; e[i] = {w, u, v}; } std::sort(e.begin(), e.end()); DSU dsu(n); int c = n; int ans = 0; std::vector<int> f(n), p(n, -1), dep(n); std::vector<int> ord;
for (auto& [w, u, v] : e) { if (!dsu.same(u, v)) { u = dsu.find(u); v = dsu.find(v); ans = (ans + (int)dsu.size(u) * dsu.size(v) * (w - 1) % MOD) % MOD; if (dsu.size(u) < dsu.size(v)) { std::swap(u, v); } c--; f[v] = w; ord.push_back(v); p[v] = u; dsu.merge(u, v); } } if (c > 2) { ans = 0; } else if (c == 2){ for (int i = 1; i < n; i++) { if (!dsu.same(i, 0)) { ans = (dsu.size(0) * dsu.size(i) % MOD) * k % MOD; break; } } } else { std::reverse(ord.begin(), ord.end()); for (auto x : ord) { dep[x] = dep[p[x]] + 1; } for (auto [w, u, v] : e) { int t = 0; while (u != v) { if (dep[u] < dep[v]) { std::swap(u, v); } t = std::max(t, f[u]); u = p[u]; } ans = (ans - (t - 1) + MOD) % MOD; } } std::cout << ans << '\n'; }
signed main() { int t = 1; std::cin >> t; while (t--) { solve(); } return 0; }
|