結果
問題 | No.1207 グラフX |
ユーザー |
![]() |
提出日時 | 2020-08-30 17:33:14 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 700 ms / 2,000 ms |
コード長 | 1,757 bytes |
コンパイル時間 | 830 ms |
コンパイル使用メモリ | 75,836 KB |
実行使用メモリ | 47,328 KB |
最終ジャッジ日時 | 2024-11-15 13:58:47 |
合計ジャッジ時間 | 24,691 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 46 |
ソースコード
//「X >= 2, z_i:相異なる」より、コスト小さい辺から追加するとき、ループができる場合をignoreできる性質が成り立つ。 //全域木作って、各辺が何回数えられるかを計算. //powmodがボトルネックでO(NlogN). #include <iostream> #include <algorithm> #include <vector> #define rep(i, n) for(i = 0; i < n; i++) #define int long long using namespace std; struct UF { int par[200000]; UF() { int i; rep(i, 200000) par[i] = i; } int root(int x) { if (par[x] == x) return x; return par[x] = root(par[x]); } void unit(int x, int y) { x = root(x); y = root(y); par[x] = y; } bool same(int x, int y) { return root(x) == root(y); } }; int powmod(int a, int n, int mod) { if (n == 0) return 1; if (n % 2 == 0) return powmod((a * a) % mod, n / 2, mod); return (a * powmod(a, n - 1, mod)) % mod; } int mod = 1000000007; int n, m, X; vector<int> et[200000]; vector<int> ec[200000]; UF uf; int sz[200000]; int dfs(int p, int v) { int i; int ret = 1; rep(i, et[v].size()) { int nv = et[v][i]; if (nv == p) continue; int res = dfs(v, nv); ret += res; } return sz[v] = ret; } int dfs2(int p, int v) { int i; int ret = 0; rep(i, et[v].size()) { int nv = et[v][i]; if (nv == p) continue; //v -- nv ret += sz[nv] * (n - sz[nv]) % mod * ec[v][i] % mod; ret %= mod; //nv以下 ret += dfs2(v, nv); ret %= mod; } return ret; } signed main() { int i, j; cin >> n >> m >> X; rep(i, m) { int x, y, z; cin >> x >> y >> z; x--; y--; if (uf.same(x, y)) continue; uf.unit(x, y); et[x].push_back(y); ec[x].push_back(powmod(X, z, mod)); et[y].push_back(x); ec[y].push_back(powmod(X, z, mod)); } dfs(-1, 0); cout << dfs2(-1, 0) << endl; return 0; }