Submission #405074

#	Time	Username	Problem	Language	Result	Execution time	Memory
405074		BERNARB01	Duathlon (APIO18_duathlon)	C++17	0 / 100	1136 ms	1048580 KiB

count_triplets

#include <bits/stdc++.h>

using namespace std;

int main() {
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n, m;
	cin >> n >> m;
	vector<vector<int>> g(n);
	for (int i = 0; i < m; i++) {
		int u, v;
		cin >> u >> v;
		--u; --v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	vector<int> sbt(n, 0);
	long long ans = 0;
	function<void(int, int)> dfs0 = [&](int u, int p) {
		sbt[u] = 1;
		for (int v : g[u]) {
			if (v != p) {
				dfs0(v, u);
				sbt[u] += sbt[v];
			}
		}
		long long sum = n - 1;
		int sbtp = n - sbt[u];
		ans += sbtp * 1LL * (sum - sbtp);
		for (int v : g[u]) {
			if (v != p) {
				ans += sbt[v] * 1LL * (sum - sbt[v]);
			}
		}
	};
	dfs0(0, -1);
	cout << ans << '\n';
	return 0;
}

• Subtask 1: 0 / 5
• Subtask 2: 0 / 11
• Subtask 3: 0 / 8
• Subtask 4: 0 / 10
• Subtask 5: 0 / 13
• Subtask 6: 0 / 15
• Subtask 7: 0 / 20
• Subtask 8: 0 / 8
• Subtask 9: 0 / 10