제출 #387781

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
387781		milleniumEeee	철인 이종 경기 (APIO18_duathlon)	C++14	5 / 100	65 ms	12812 KiB

count_triplets

#include <bits/stdc++.h>
#define fastInp ios_base::sync_with_stdio(0); cin.tie(0);
#define pii pair<int, int>
#define fr first
#define sc second
#define pb push_back
#define szof(s) (int)s.size()
#define all(s) s.begin(), s.end()
#define mk make_pair
#define int long long
using namespace std;

const int MAXN = (int)1e5 + 5;

vector <int> g[MAXN];
int n, m;
int S, C, F;
bool used[MAXN];

bool dfs(int v) {
	used[v] = 1;
	if (v == F) {
		used[v] = 0;
		if (!used[C]) {
			return false;
		} else {
			return true;
		}
	}
	bool found = false;
	for (int to : g[v]) {
		if (!used[to]) {
			found |= dfs(to);
			if (found) {
				used[v] = 0;
				return true;
			}
		}
	}
	used[v] = 0;
	return false;
}

void subtask1() {
	int ans = 0;
	for (int s = 1; s <= n; s++) {
		for (int c = 1; c <= n; c++) {
			for (int f = 1; f <= n; f++) {
				if (s != c && s != f && c != f) {
					S = s;
					C = c;
					F = f;
					if (dfs(s)) {
						ans++;
					}
				}
			}
		}
	}
	cout << ans << endl;	
	exit(0);
}


int color[MAXN];

bool cycle(int v) {
	color[v] = 1;
	bool f = false;
	for (int to : g[v]) {
		if (color[to] == 0) {
			f |= cycle(to);
		}
		if (color[to] == 1) {
			f = 1;
		}
	}
	color[v] = 2;
	return f;
}

bool us[MAXN];
void get_sz(int v, int &sz) {
	us[v] = 1;
	sz++;
	for (int to : g[v]) {
		if (!us[to]) {
			get_sz(to, sz);
		}
	}
}

void subtask3() {
	for (int i = 1; i <= n; i++) {
		if (szof(g[i]) > 2) {
			return;
		}
	}
	bool flag = 0;
	for (int i = 1; i <= n; i++) {
		if (szof(g[i]) == 1) {
			flag = 1;
		}
	}
	if (!cycle(1)) {
		int sum = 0;
		for (int i = 1; i <= n; i++) {
			sum += (i - 1) * (n - i);
		}
		cout << sum << endl;
	} else {
		cout << n * (n - 1) * (n - 2) << endl;
	}
	exit(0);
}

bool was[MAXN];
int cnt = 0;
void check(int v) {
	was[v] = 1;
	cnt++;
	for (int to : g[v]) {
		if (!was[to]) {
			check(to);
		}
	}
}

signed main() {
	fastInp;
	cin >> n >> m;
	for (int i = 1; i <= m; i++) {
		int u, v;
		cin >> u >> v;
		g[u].pb(v);
		g[v].pb(u);
	}
	// могут быть несколько компонент
	if (n <= 10 && m <= 100) {
		subtask1();
	}
	subtask3();
}
/*
4 3
1 2
2 3
3 4
*/

컴파일 시 표준 에러 (stderr) 메시지

count_triplets.cpp: In function 'void subtask3()':
count_triplets.cpp:99:7: warning: variable 'flag' set but not used [-Wunused-but-set-variable]
   99 |  bool flag = 0;
      |       ^~~~

• Subtask 1: 5 / 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