Submission #172570

#	Submission time	Handle	Problem	Language	Result	Execution time	Memory
172570	2020-01-02T03:33:04 Z	eggag32	Duathlon (APIO18_duathlon)	C++17	36 / 100	155 ms	12108 KB

count_triplets

#pragma GCC optimize ("O3")
#pragma GCC target ("sse4")
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef long double ld;
typedef vector<int> vi;
typedef pair<int, int> pi;
#define debug(x) cerr << #x << ": " << x << endl;
#define debug2(x, y) debug(x) debug(y);
#define repn(i, a, b) for(int i = (int)(a); i < (int)(b); i++)
#define rep(i, a) for(int i = 0; i < (int)(a); i++)
#define all(v) v.begin(), v.end() 
#define mp make_pair
#define pb push_back
#define lb lower_bound
#define ub upper_bound
#define fi first
#define se second
#define sq(x) ((x) * (x))

template<class T> T gcd(T a, T b){ return ((b == 0) ? a : gcd(b, a % b)); }

ll n, m;
vi vis2;
vi g[100005];

bool dfs(int cur, int mid, int fin, bool fnd, vi vis){
	vis[cur] = 1;
	if(cur == mid) fnd = 1;
	if(cur == fin){
		if(fnd) return 1;
		return 0;
	}
	int ret = 0;
	for(int u : g[cur]) if(!vis[u]){
		ret |= dfs(u, mid, fin, fnd, vis);
		if(ret) break;
	}
	return ret;
}

bool check(int st, int mid, int fin){
	vis2.assign(n, 0);
	return dfs(st, mid, fin, 0, vis2);
}

ll ans1 = 0;
int vis1[100005];
int vis3[100005];
int ind[100005];
int last[100005];
ll cn[100005];
ll d[100005];
int id = 0;
ll cc = 0;

void dfs1(int cur){
	//initial traversal to compute the subtree sizes
	vis1[cur] = 1;
	ind[cur] = id;
	last[id] = cur;
	cc++;
	d[cur] = 1;
	for(int u : g[cur]) if(!vis1[u]){
		dfs1(u);
		d[cur] += d[u];
	}
}	

void dfs2(int cur){
	vis3[cur] = 1;
	ll sm = 0;
	for(int u : g[cur]) if(!vis3[u]){
		ans1 += (ll)(d[u] * (cn[ind[cur]] - d[u] - 1));
		sm += d[u];
	}
	ans1 += (ll)((cn[ind[cur]] - d[cur]) * sm);
	for(int u : g[cur]) if(!vis3[u]) dfs2(u);
}

ll ans2 = 0;
int vis4[100005];
ll cnt1[2];

bool dfs4(int cur, int prev){
	vis4[cur] = 1;
	if(prev != -1){
		cnt1[1] += cnt1[0];
		cnt1[0]++;
	}
	int ret = 0;
	for(int u : g[cur]) if(u != prev && (ind[u] == ind[cur])){
		if(vis4[u]) return 1;
		ret |= dfs4(u, cur);
	}
	return ret;
}

int main(){
	ios_base::sync_with_stdio(false);
	cin.tie(0);
	//freopen("1.out", "r", stdin);
	//freopen("output.out", "w", stdout);
	cin >> n >> m;
	int mx = 0;
	rep(i, m){
		int a, b;
		cin >> a >> b;
		a--, b--;
		g[a].pb(b);
		g[b].pb(a);
		mx = max({mx, (int)g[a].size(), (int)g[b].size()});
	}
	if(n <= 10 && m <= 100){
		//first subtask
		ll ans = 0;
		rep(i, n) repn(k, i + 1, n) rep(j, n) if(j != i && j != k) if(check(i, j, k)) ans += 2;
		cout << ans << endl;
	}
	else{
		//subtasks 3, 4, 5
		memset(vis1, 0, sizeof(vis1));
		memset(vis3, 0, sizeof(vis3));
		memset(vis4, 0, sizeof(vis4));
		memset(d, 0, sizeof(d));
		rep(i, n) if(!vis1[i]){
			cc = 0;
			dfs1(i);
			cn[id] = cc;
			id++;
		}
		if(mx == 2){
			int f = 0;
			vi cyc(100005, 0);
			rep(i, id){
				if(dfs4(last[i], -1)) cyc[i] = 1, f = 1;
				else{
					for(ll j = 2; j < cn[i]; j++) ans2 += (ll)(2 * (j - 1) * (cn[i] - j) * 1LL);
				}
			}
			if(!f) goto g;
			rep(i, id) if(cyc[i]){
				ans2 += ((cn[i]) * (cn[i] - 1) * (cn[i] - 2));
			}
			cout << ans2 << endl;
			return 0;
		}
g:;
		rep(i, n) if(!vis3[i]) dfs2(i);
		cout << ans1 << endl;
	}
	return 0;
}
/*
Things to look out for:
	- Integer overflows
	- Array bounds
	- Special cases
Be careful!
*/

Subtask #1
5.0 / 5.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	4 ms	2680 KB	Output is correct
2	Correct	4 ms	2680 KB	Output is correct
3	Correct	4 ms	2680 KB	Output is correct
4	Correct	4 ms	2680 KB	Output is correct
5	Correct	4 ms	2680 KB	Output is correct
6	Correct	4 ms	2680 KB	Output is correct
7	Correct	4 ms	2680 KB	Output is correct
8	Correct	6 ms	2680 KB	Output is correct
9	Correct	4 ms	2680 KB	Output is correct
10	Correct	4 ms	2680 KB	Output is correct
11	Correct	4 ms	2680 KB	Output is correct
12	Correct	5 ms	2680 KB	Output is correct
13	Correct	5 ms	2680 KB	Output is correct
14	Correct	5 ms	2680 KB	Output is correct
15	Correct	4 ms	2680 KB	Output is correct
16	Correct	4 ms	2680 KB	Output is correct
17	Correct	4 ms	2680 KB	Output is correct
18	Correct	5 ms	2680 KB	Output is correct
19	Correct	4 ms	2680 KB	Output is correct
20	Correct	4 ms	2680 KB	Output is correct
21	Correct	4 ms	2680 KB	Output is correct
22	Correct	5 ms	2680 KB	Output is correct
23	Correct	4 ms	2680 KB	Output is correct
24	Correct	4 ms	2680 KB	Output is correct
25	Correct	5 ms	2808 KB	Output is correct
26	Correct	5 ms	2684 KB	Output is correct
27	Correct	5 ms	2680 KB	Output is correct
28	Correct	5 ms	2680 KB	Output is correct
29	Correct	5 ms	2680 KB	Output is correct
30	Correct	4 ms	2680 KB	Output is correct
31	Correct	4 ms	2680 KB	Output is correct
32	Correct	4 ms	2680 KB	Output is correct
33	Correct	4 ms	2680 KB	Output is correct
34	Correct	4 ms	2680 KB	Output is correct
35	Correct	4 ms	2808 KB	Output is correct

Subtask #2
0 / 11.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	4 ms	2680 KB	Output is correct
2	Correct	4 ms	2680 KB	Output is correct
3	Correct	4 ms	2680 KB	Output is correct
4	Correct	4 ms	2680 KB	Output is correct
5	Correct	4 ms	2680 KB	Output is correct
6	Correct	4 ms	2680 KB	Output is correct
7	Correct	4 ms	2680 KB	Output is correct
8	Correct	6 ms	2680 KB	Output is correct
9	Correct	4 ms	2680 KB	Output is correct
10	Correct	4 ms	2680 KB	Output is correct
11	Correct	4 ms	2680 KB	Output is correct
12	Correct	5 ms	2680 KB	Output is correct
13	Correct	5 ms	2680 KB	Output is correct
14	Correct	5 ms	2680 KB	Output is correct
15	Correct	4 ms	2680 KB	Output is correct
16	Correct	4 ms	2680 KB	Output is correct
17	Correct	4 ms	2680 KB	Output is correct
18	Correct	5 ms	2680 KB	Output is correct
19	Correct	4 ms	2680 KB	Output is correct
20	Correct	4 ms	2680 KB	Output is correct
21	Correct	4 ms	2680 KB	Output is correct
22	Correct	5 ms	2680 KB	Output is correct
23	Correct	4 ms	2680 KB	Output is correct
24	Correct	4 ms	2680 KB	Output is correct
25	Correct	5 ms	2808 KB	Output is correct
26	Correct	5 ms	2684 KB	Output is correct
27	Correct	5 ms	2680 KB	Output is correct
28	Correct	5 ms	2680 KB	Output is correct
29	Correct	5 ms	2680 KB	Output is correct
30	Correct	4 ms	2680 KB	Output is correct
31	Correct	4 ms	2680 KB	Output is correct
32	Correct	4 ms	2680 KB	Output is correct
33	Correct	4 ms	2680 KB	Output is correct
34	Correct	4 ms	2680 KB	Output is correct
35	Correct	4 ms	2808 KB	Output is correct
36	Correct	6 ms	4732 KB	Output is correct
37	Correct	6 ms	4628 KB	Output is correct
38	Incorrect	6 ms	4728 KB	Output isn't correct
39	Halted	0 ms	0 KB	-

Subtask #3
8.0 / 8.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	105 ms	12108 KB	Output is correct
2	Correct	107 ms	12024 KB	Output is correct
3	Correct	99 ms	10360 KB	Output is correct
4	Correct	108 ms	11256 KB	Output is correct
5	Correct	149 ms	9848 KB	Output is correct
6	Correct	155 ms	10492 KB	Output is correct
7	Correct	154 ms	9336 KB	Output is correct
8	Correct	94 ms	9848 KB	Output is correct
9	Correct	96 ms	8952 KB	Output is correct
10	Correct	90 ms	10488 KB	Output is correct
11	Correct	73 ms	9976 KB	Output is correct
12	Correct	67 ms	9848 KB	Output is correct
13	Correct	69 ms	9720 KB	Output is correct
14	Correct	66 ms	9692 KB	Output is correct
15	Correct	48 ms	9336 KB	Output is correct
16	Correct	47 ms	9336 KB	Output is correct
17	Correct	9 ms	6264 KB	Output is correct
18	Correct	9 ms	6264 KB	Output is correct
19	Correct	9 ms	6136 KB	Output is correct
20	Correct	9 ms	6264 KB	Output is correct
21	Correct	11 ms	6648 KB	Output is correct
22	Correct	9 ms	6136 KB	Output is correct

Subtask #4
10.0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	4728 KB	Output is correct
2	Correct	6 ms	4728 KB	Output is correct
3	Correct	6 ms	4728 KB	Output is correct
4	Correct	7 ms	5112 KB	Output is correct
5	Correct	6 ms	4728 KB	Output is correct
6	Correct	6 ms	4756 KB	Output is correct
7	Correct	7 ms	5112 KB	Output is correct
8	Correct	6 ms	4728 KB	Output is correct
9	Correct	6 ms	4728 KB	Output is correct
10	Correct	6 ms	4728 KB	Output is correct
11	Correct	6 ms	4728 KB	Output is correct
12	Correct	6 ms	4728 KB	Output is correct
13	Correct	6 ms	4728 KB	Output is correct
14	Correct	6 ms	4728 KB	Output is correct
15	Correct	1 ms	4732 KB	Output is correct
16	Correct	6 ms	4664 KB	Output is correct
17	Correct	6 ms	4728 KB	Output is correct
18	Correct	6 ms	4728 KB	Output is correct
19	Correct	6 ms	4728 KB	Output is correct
20	Correct	6 ms	4728 KB	Output is correct

Subtask #5
13.0 / 13.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	85 ms	8312 KB	Output is correct
2	Correct	72 ms	8312 KB	Output is correct
3	Correct	78 ms	8312 KB	Output is correct
4	Correct	77 ms	8312 KB	Output is correct
5	Correct	79 ms	8312 KB	Output is correct
6	Correct	123 ms	12024 KB	Output is correct
7	Correct	93 ms	9336 KB	Output is correct
8	Correct	92 ms	9080 KB	Output is correct
9	Correct	97 ms	8696 KB	Output is correct
10	Correct	88 ms	8380 KB	Output is correct
11	Correct	107 ms	8312 KB	Output is correct
12	Correct	75 ms	8440 KB	Output is correct
13	Correct	69 ms	8312 KB	Output is correct
14	Correct	72 ms	8440 KB	Output is correct
15	Correct	60 ms	8312 KB	Output is correct
16	Correct	39 ms	8056 KB	Output is correct
17	Correct	48 ms	8560 KB	Output is correct
18	Correct	49 ms	8560 KB	Output is correct
19	Correct	51 ms	8556 KB	Output is correct
20	Correct	49 ms	8544 KB	Output is correct

Subtask #6
0 / 15.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	6 ms	4728 KB	Output is correct
2	Correct	2 ms	4728 KB	Output is correct
3	Incorrect	7 ms	4728 KB	Output isn't correct
4	Halted	0 ms	0 KB	-

Subtask #7
0 / 20.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	78 ms	8292 KB	Output is correct
2	Correct	79 ms	8184 KB	Output is correct
3	Incorrect	76 ms	8184 KB	Output isn't correct
4	Halted	0 ms	0 KB	-

Subtask #8
0 / 8.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	4 ms	2680 KB	Output is correct
2	Correct	4 ms	2680 KB	Output is correct
3	Correct	4 ms	2680 KB	Output is correct
4	Correct	4 ms	2680 KB	Output is correct
5	Correct	4 ms	2680 KB	Output is correct
6	Correct	4 ms	2680 KB	Output is correct
7	Correct	4 ms	2680 KB	Output is correct
8	Correct	6 ms	2680 KB	Output is correct
9	Correct	4 ms	2680 KB	Output is correct
10	Correct	4 ms	2680 KB	Output is correct
11	Correct	4 ms	2680 KB	Output is correct
12	Correct	5 ms	2680 KB	Output is correct
13	Correct	5 ms	2680 KB	Output is correct
14	Correct	5 ms	2680 KB	Output is correct
15	Correct	4 ms	2680 KB	Output is correct
16	Correct	4 ms	2680 KB	Output is correct
17	Correct	4 ms	2680 KB	Output is correct
18	Correct	5 ms	2680 KB	Output is correct
19	Correct	4 ms	2680 KB	Output is correct
20	Correct	4 ms	2680 KB	Output is correct
21	Correct	4 ms	2680 KB	Output is correct
22	Correct	5 ms	2680 KB	Output is correct
23	Correct	4 ms	2680 KB	Output is correct
24	Correct	4 ms	2680 KB	Output is correct
25	Correct	5 ms	2808 KB	Output is correct
26	Correct	5 ms	2684 KB	Output is correct
27	Correct	5 ms	2680 KB	Output is correct
28	Correct	5 ms	2680 KB	Output is correct
29	Correct	5 ms	2680 KB	Output is correct
30	Correct	4 ms	2680 KB	Output is correct
31	Correct	4 ms	2680 KB	Output is correct
32	Correct	4 ms	2680 KB	Output is correct
33	Correct	4 ms	2680 KB	Output is correct
34	Correct	4 ms	2680 KB	Output is correct
35	Correct	4 ms	2808 KB	Output is correct
36	Correct	6 ms	4732 KB	Output is correct
37	Correct	6 ms	4628 KB	Output is correct
38	Incorrect	6 ms	4728 KB	Output isn't correct
39	Halted	0 ms	0 KB	-

Subtask #9
0 / 10.0

#	Verdict	Execution time	Memory	Grader output
1	Correct	4 ms	2680 KB	Output is correct
2	Correct	4 ms	2680 KB	Output is correct
3	Correct	4 ms	2680 KB	Output is correct
4	Correct	4 ms	2680 KB	Output is correct
5	Correct	4 ms	2680 KB	Output is correct
6	Correct	4 ms	2680 KB	Output is correct
7	Correct	4 ms	2680 KB	Output is correct
8	Correct	6 ms	2680 KB	Output is correct
9	Correct	4 ms	2680 KB	Output is correct
10	Correct	4 ms	2680 KB	Output is correct
11	Correct	4 ms	2680 KB	Output is correct
12	Correct	5 ms	2680 KB	Output is correct
13	Correct	5 ms	2680 KB	Output is correct
14	Correct	5 ms	2680 KB	Output is correct
15	Correct	4 ms	2680 KB	Output is correct
16	Correct	4 ms	2680 KB	Output is correct
17	Correct	4 ms	2680 KB	Output is correct
18	Correct	5 ms	2680 KB	Output is correct
19	Correct	4 ms	2680 KB	Output is correct
20	Correct	4 ms	2680 KB	Output is correct
21	Correct	4 ms	2680 KB	Output is correct
22	Correct	5 ms	2680 KB	Output is correct
23	Correct	4 ms	2680 KB	Output is correct
24	Correct	4 ms	2680 KB	Output is correct
25	Correct	5 ms	2808 KB	Output is correct
26	Correct	5 ms	2684 KB	Output is correct
27	Correct	5 ms	2680 KB	Output is correct
28	Correct	5 ms	2680 KB	Output is correct
29	Correct	5 ms	2680 KB	Output is correct
30	Correct	4 ms	2680 KB	Output is correct
31	Correct	4 ms	2680 KB	Output is correct
32	Correct	4 ms	2680 KB	Output is correct
33	Correct	4 ms	2680 KB	Output is correct
34	Correct	4 ms	2680 KB	Output is correct
35	Correct	4 ms	2808 KB	Output is correct
36	Correct	6 ms	4732 KB	Output is correct
37	Correct	6 ms	4628 KB	Output is correct
38	Incorrect	6 ms	4728 KB	Output isn't correct
39	Halted	0 ms	0 KB	-

Submission #172570

Compilation message