树上求和(DFS)

在这里插入图片描述
在这里插入图片描述
解题思路:
一条边左边节点个数乘以右边顶点个数等于这条边出现的次数

#include<bits/stdc++.h>
using namespace std;
#define ll long long
vector<int> e[100010];
ll n,ans[100010],cnt;
bool vis[100010];
ll dfs(int x)
{
	ll t=0;
	for(auto c:e[x])
	{
		if(!vis[c])
		{
			vis[c]=true;
			ll temp=dfs(c)+1;
			t+=temp;
			ans[++cnt]=temp*(n-temp);
		}
	}
	return t;
} 
int main()
{
	cin>>n;
	for(int i=0;i<n-1;++i)
	{
		int x,y;
		cin>>x>>y;
		e[x].push_back(y);
		e[y].push_back(x);
	}
	vis[1]=true;
	dfs(1);
	sort(ans+1,ans+n);
	long long res=0;
	for(int i=1,j=n-1;i<n;++i,--j)
	{
		res+=ans[i]*j;
	}
	cout<<res<<endl;
	return 0;
} 

注意开long long,不开long long后面数据会溢出

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值