Submission #764080

#	Time	Username	Problem	Language	Result	Execution time	Memory
764080		OrazB	Duathlon (APIO18_duathlon)	C++14	23 / 100	925 ms	1048576 KiB

count_triplets

#include <bits/stdc++.h>
#include <ext/pb_ds/assoc_container.hpp>
#include <ext/pb_ds/tree_policy.hpp>
#include <functional>
using namespace __gnu_pbds;
using namespace std;
typedef tree<int, null_type, less<int>, rb_tree_tag,
             tree_order_statistics_node_update>
    ordered_set;
//Dijkstra->set
//set.find_by_order(x) x-position value
//set.order_of_key(x) number of strictly less elements don't need *set.??
#define N 2000005
#define wr cout << "Continue debugging\n";
#define all(x) (x).begin(), (x).end()
#define ll long long int
#define pii pair <int, int>
#define pb push_back
#define ff first
#define ss second

bool vis[N];
ll sub[N], sum;
vector<int> E[N];
vector<int> vec;

void dfs(int nd){
	vis[nd] = sub[nd] = 1;
	for (auto i : E[nd]){
		if (!vis[i]){
			dfs(i);
			sub[nd] += sub[i];
		} 
	}
}
void dfsz(int nd, int pr, ll sz){
	sum += (sz-sub[nd])*(sub[nd]-1)*2;
	for (auto i : E[nd]){
		if (i == pr) continue;
		dfsz(i,nd,sz);
		sum += (sub[nd]-sub[i]-1)*sub[i];
	}
}

int main ()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	int n, m;
	cin >> n >> m;
	while(m--){
		int u, v;
		cin >> u >> v;
		E[u].pb(v);
		E[v].pb(u);
	}
	for (int i = 1; i <= n; i++){
		if (!vis[i]){
			dfs(i);
			dfsz(i, 0, sub[i]);
		}
	}
	cout << sum << '\n';
}	

Compilation message (stderr)

count_triplets.cpp: In function 'void dfs(int)':
count_triplets.cpp:28:20: warning: suggest parentheses around assignment used as truth value [-Wparentheses]
   28 |  vis[nd] = sub[nd] = 1;
      |            ~~~~~~~~^~~

• 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