#566146 (C++14) No.1207 グラフX

提出ソース

結果

問題	No.1207 グラフX
ユーザー	👑 Nachia
提出日時	2020-10-12 17:48:24
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	321 ms / 2,000 ms
コード長	1,265 bytes
コンパイル時間	684 ms
コンパイル使用メモリ	76,400 KB
実行使用メモリ	36,968 KB
最終ジャッジ日時	2023-09-27 23:46:59
合計ジャッジ時間	13,300 ms
ジャッジサーバーID （参考情報）	judge15 / judge12

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	259 ms 20,536 KB
testcase_01	AC	253 ms 20,380 KB
testcase_02	AC	255 ms 20,876 KB
testcase_03	AC	259 ms 20,388 KB
testcase_04	AC	261 ms 20,824 KB
testcase_05	AC	317 ms 36,880 KB
testcase_06	AC	306 ms 36,936 KB
testcase_07	AC	321 ms 36,956 KB
testcase_08	AC	229 ms 15,196 KB
testcase_09	AC	228 ms 17,504 KB
testcase_10	AC	307 ms 29,884 KB
testcase_11	AC	306 ms 36,968 KB
testcase_12	AC	216 ms 16,056 KB
testcase_13	AC	151 ms 10,044 KB
testcase_14	AC	255 ms 19,536 KB
testcase_15	AC	255 ms 17,012 KB
testcase_16	AC	154 ms 10,576 KB
testcase_17	AC	196 ms 14,704 KB
testcase_18	AC	153 ms 14,900 KB
testcase_19	AC	217 ms 12,308 KB
testcase_20	AC	259 ms 20,144 KB
testcase_21	AC	28 ms 8,088 KB
testcase_22	AC	182 ms 14,888 KB
testcase_23	AC	194 ms 15,792 KB
testcase_24	AC	125 ms 13,808 KB
testcase_25	AC	250 ms 19,688 KB
testcase_26	AC	201 ms 16,864 KB
testcase_27	AC	246 ms 18,496 KB
testcase_28	AC	240 ms 17,428 KB
testcase_29	AC	237 ms 18,556 KB
testcase_30	AC	130 ms 11,964 KB
testcase_31	AC	127 ms 9,320 KB
testcase_32	AC	100 ms 12,372 KB
testcase_33	AC	120 ms 12,172 KB
testcase_34	AC	233 ms 16,696 KB
testcase_35	AC	28 ms 8,236 KB
testcase_36	AC	233 ms 17,304 KB
testcase_37	AC	207 ms 15,456 KB
testcase_38	AC	57 ms 9,612 KB
testcase_39	AC	120 ms 12,516 KB
testcase_40	AC	112 ms 8,360 KB
testcase_41	AC	187 ms 12,504 KB
testcase_42	AC	3 ms 8,056 KB
testcase_43	AC	3 ms 7,964 KB
testcase_44	AC	3 ms 8,092 KB
testcase_45	AC	219 ms 20,524 KB
testcase_46	AC	227 ms 20,056 KB
testcase_47	AC	227 ms 20,112 KB
testcase_48	AC	229 ms 20,124 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <iostream>
#include <vector>
using namespace std;
using LL = long long;
using ULL = unsigned long long;
#define rep(i,n) for(int i=0; i<(n); i++)

struct dsu{
	struct Node{ int v, z; };
	vector<Node> V;
	dsu(int N){ V.resize(N); rep(i, N) V[i] = { i, 1 }; }
	int root(int a){
		if (V[a].v == a) return a;
		return V[a].v = root(V[a].v);
	}
	int size(int a){ return V[root(a)].z; }
	void unite(int a, int b){
		a = root(a); b = root(b);
		V[a].z += V[b].z;
		V[b].v = a;
	}
};

const ULL M = 1000000007;

ULL powm(ULL a, ULL i){
	if (i == 0) return 1;
	ULL res = powm(a * a % M, i / 2);
	if (i % 2 == 1) res = res * a % M;
	return res;
}

int N, nE;
ULL X;
vector<pair<int, ULL>> E[200000];
ULL Z[200000];
ULL ans = 0;

void dfs(int p, int pre = -1){
	Z[p] = 1;
	for (auto e : E[p]){
		if (e.first == pre) continue;
		dfs(e.first, p);
		Z[p] += Z[e.first];
		ULL tmp = e.second;
		tmp = tmp * Z[e.first] % M * (N - Z[e.first]) % M;
		ans += tmp;
		ans %= M;
	}
}

int main(){
	cin >> N >> nE;
	cin >> X;
	dsu G(N);

	rep(i, nE){
		int u, v; ULL z; cin >> u >> v >> z; u--; v--;
		if (G.root(u) == G.root(v)) continue;
		z = powm(X, z);
		E[u].push_back({ v, z });
		E[v].push_back({ u, z });
		G.unite(u, v);
	}
	dfs(0);

	cout << ans << endl;

	return 0;
}

yukicoder

結果

テストケース

ソースコード