Submission #71939

#	Time	Username	Problem	Language	Result	Execution time	Memory
71939		KLPP	철인 이종 경기 (APIO18_duathlon)	C++14	0 / 100	199 ms	36592 KiB

count_triplets

#include<iostream>
#include<stdio.h>
#include<vector>
#include<queue>
#include<algorithm>

using namespace std;
typedef long long int lld;
lld DP[1000000];
bool visited[1000000];
vector<int>nei[1000000];
lld sumdist[1000000];
lld ans;
void DFS(int node){
	visited[node]=true;
	lld sum=0;
	lld square=0;
	lld totaldist=0;
	for(int i=0;i<nei[node].size();i++){
		int v=nei[node][i];
		if(!visited[v]){
			DFS(v);
			sum+=DP[v];
			square+=DP[v]*DP[v];
			totaldist+=sumdist[v]+DP[v]-1;
		}
	}
	DP[node]=sum+1;
	sumdist[node]=totaldist;
	ans+=sum*sum-square;
	ans+=2*totaldist;
}
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=0;i<m;i++){
		int x,y;
		cin>>x>>y;
		x--;y--;
		nei[x].push_back(y);
		nei[y].push_back(x);
	}
	ans=0;
	for(int i=0;i<n;i++){
		DP[i]=-1;
		visited[i]=false;
	}
	for(int i=0;i<n;i++){
		if(!visited[i]){
			DFS(i);
		}
	}//for(int i=0;i<n;i++)cout<<sumdist[i]<<endl;
	cout<<ans<<endl;

	return 0;	
}

Compilation message (stderr)

count_triplets.cpp: In function 'void DFS(int)':
count_triplets.cpp:19:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<nei[node].size();i++){
              ~^~~~~~~~~~~~~~~~~

• 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