HDU-4705 树形DP

最新推荐文章于 2022-07-03 23:18:30 发布

丶聂小白

最新推荐文章于 2022-07-03 23:18:30 发布

阅读量448

点赞数

分类专栏：集训树形DP DP ACM 文章标签： acm hdu

本文链接：https://blog.csdn.net/q295657451/article/details/38899221

版权

ACM 同时被 3 个专栏收录

129 篇文章 0 订阅

订阅专栏

集训

47 篇文章 0 订阅

订阅专栏

33 篇文章 2 订阅

订阅专栏

C++能过，G++会爆栈，+了扩栈头文件也是这样

#pragma comment(linker, "/STACK:1024000000,1024000000")
#include <iostream>
#include <cstdio>
#include <cmath>
#include <queue>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=110000;
struct Edge
{
	int v,next;
}edge[maxn*2];
int head[maxn];
int sum[maxn];
int num;
int n;
long long ans;
void addedge(int u,int v)
{
	edge[num].v=v;
	edge[num].next=head[u];
	head[u]=num++;
	edge[num].v=u;
	edge[num].next=head[v];
	head[v]=num++;
}
void dfs(int root,int father)
{
	sum[root]=1;
	int temp=0;
	for(int i=head[root];i!=-1;i=edge[i].next)
	{
		int v=edge[i].v;	
		if(v==father)
		{
			continue;
		}
		dfs(v,root);
		ans+=(long long)temp*sum[v];
		sum[root]+=sum[v];
		temp+=sum[v];
	}
	ans+=(long long)temp*(n-sum[root]);
}
int main()
{
	int u,v;
	while(scanf("%d",&n)!=EOF)
	{
		num=0;
		ans=0;
		memset(head,-1,sizeof(head));
		for(int i=1;i<n;i++)
		{
			scanf("%d%d",&u,&v);
			addedge(u,v);
		}
		dfs(1,-1);
		long long s=(long long)n*(n-1)*(n-2)/6;//N个选3个 
		printf("%I64d\n",s-ans);
	}
	return 0;
}