結果

問題 No.1207 グラフX
ユーザー e869120e869120
提出日時 2021-02-26 08:32:25
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 667 ms / 2,000 ms
コード長 1,659 bytes
コンパイル時間 906 ms
コンパイル使用メモリ 77,400 KB
実行使用メモリ 32,720 KB
最終ジャッジ日時 2024-10-01 22:34:56
合計ジャッジ時間 24,409 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 577 ms
23,816 KB
testcase_01 AC 578 ms
23,816 KB
testcase_02 AC 595 ms
23,968 KB
testcase_03 AC 597 ms
24,288 KB
testcase_04 AC 604 ms
23,948 KB
testcase_05 AC 659 ms
32,588 KB
testcase_06 AC 667 ms
32,592 KB
testcase_07 AC 659 ms
32,460 KB
testcase_08 AC 432 ms
20,812 KB
testcase_09 AC 490 ms
22,204 KB
testcase_10 AC 627 ms
28,192 KB
testcase_11 AC 637 ms
32,720 KB
testcase_12 AC 445 ms
20,568 KB
testcase_13 AC 221 ms
16,468 KB
testcase_14 AC 572 ms
23,696 KB
testcase_15 AC 486 ms
21,420 KB
testcase_16 AC 222 ms
16,708 KB
testcase_17 AC 370 ms
19,540 KB
testcase_18 AC 333 ms
19,452 KB
testcase_19 AC 332 ms
18,108 KB
testcase_20 AC 583 ms
23,564 KB
testcase_21 AC 36 ms
14,188 KB
testcase_22 AC 363 ms
19,476 KB
testcase_23 AC 411 ms
20,424 KB
testcase_24 AC 290 ms
18,616 KB
testcase_25 AC 576 ms
23,432 KB
testcase_26 AC 443 ms
21,096 KB
testcase_27 AC 531 ms
22,780 KB
testcase_28 AC 496 ms
22,252 KB
testcase_29 AC 505 ms
22,736 KB
testcase_30 AC 245 ms
18,380 KB
testcase_31 AC 186 ms
16,716 KB
testcase_32 AC 221 ms
17,312 KB
testcase_33 AC 231 ms
18,636 KB
testcase_34 AC 464 ms
21,292 KB
testcase_35 AC 40 ms
14,456 KB
testcase_36 AC 471 ms
21,928 KB
testcase_37 AC 397 ms
19,948 KB
testcase_38 AC 109 ms
15,744 KB
testcase_39 AC 244 ms
18,768 KB
testcase_40 AC 136 ms
15,032 KB
testcase_41 AC 309 ms
18,052 KB
testcase_42 AC 5 ms
15,824 KB
testcase_43 AC 5 ms
14,560 KB
testcase_44 AC 5 ms
15,816 KB
testcase_45 AC 551 ms
23,808 KB
testcase_46 AC 573 ms
24,064 KB
testcase_47 AC 564 ms
23,932 KB
testcase_48 AC 564 ms
23,812 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;

long long mod = 1000000007;

class UnionFind {
public:
	vector<int> par;

	void init(int sz) {
		par.resize(sz, -1);
	}
	int root(int pos) {
		if (par[pos] == -1) return pos;
		par[pos] = root(par[pos]);
		return par[pos];
	}
	void unite(int u, int v) {
		u = root(u); v = root(v);
		if (u != v) par[u] = v;
	}
	bool same(int u, int v) {
		u = root(u); v = root(v);
		if (u == v) return true;
		return false;
	}
};

long long modpow(long long a, long long b, long long m) {
	long long p = 1, q = a;
	for (int i = 0; i < 62; i++) {
		if ((b / (1LL << i)) % 2LL == 1) {
			p *= q; p %= m;
		}
		q *= q; q %= m;
	}
	return p;
}

// 入力
long long N, M, X;
long long A[1 << 18], B[1 << 18], C[1 << 18];

// 木の情報
UnionFind UF;
vector<int> G[1 << 18];
int dp[1 << 18];
bool used[1 << 18];

void dfs(int pos) {
	dp[pos] = 1;
	for (int i : G[pos]) {
		if (dp[i] >= 1) continue;
		dfs(i);
		dp[pos] += dp[i];
	}
}

int main() {
	// Step #1. 入力
	cin >> N >> M >> X;
	for (int i = 1; i <= M; i++) cin >> A[i] >> B[i] >> C[i];

	// Step #2. UF
	UF.init(N + 2);
	for (int i = 1; i <= M; i++) {
		if (UF.same(A[i], B[i]) == false) {
			UF.unite(A[i], B[i]);
			G[A[i]].push_back(B[i]);
			G[B[i]].push_back(A[i]);
			used[i] = true;
		}
	}

	// Step #3. 答えを求める
	long long Answer = 0;
	dfs(1);
	for (int i = 1; i <= M; i++) {
		if (used[i] == false) continue;
		long long cost = modpow(X, C[i], mod);
		long long t1 = min(dp[A[i]], dp[B[i]]);
		long long t2 = N - t1;
		Answer += (t1 * t2 % mod) * cost % mod;
		Answer %= mod;
	}
	cout << Answer << endl;
	return 0;
}
0