結果
問題 | 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 longusing 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 -- nvret += 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;}