제출 #983011

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
983011		vjudge1	철인 이종 경기 (APIO18_duathlon)	C++17	23 / 100	38 ms	15964 KiB

count_triplets

#include <bits/stdc++.h>
#define int long long
using namespace std;
int n, m, i, j, x, y, ans, sz[200001], rt[200001];
bool u[200001];
vector<int> g[200001], v;

void dfs(int x, int r) {
	sz[x] = 1;
	u[x] = 1;
	rt[x] = r;
	for (auto y : g[x]) {
		if (u[y]) continue;
		dfs(y, r);
		sz[x] += sz[y];
	}
}

signed main() {
	cin.tie(0)->sync_with_stdio(0);
	#ifdef LOCAL
	freopen("input.txt", "r", stdin);
	#endif
	cin >> n >> m;
	while (m--) {
		cin >> x >> y;
		g[x].push_back(y);
		g[y].push_back(x);
	}
	v.push_back(1);
	dfs(1, 1);
	for (i = 1; i <= n; i++) {
		if (!u[i]) {
			v.push_back(i);
			dfs(i, i);
		}
	}
	for (int i : v) {
		ans -= (sz[i]) * (sz[i] - 1); // + (x, x)
	}
	for (i = 1; i <= n; i++) {
		//cout << i << " " << rt[i] << " " << sz[i] << " " << sz[rt[i]] << " " << sz[rt[i]] - sz[i] << "\n";
		ans += 2 * (sz[i] - 1) * (sz[rt[i]] - sz[i]);
		ans += 2 * (sz[rt[i]] - sz[i]); // - (x, x)
	}
	cout << ans;
}

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