codeforces E. Sergey and Subway(边权和变形||dfs)

博客围绕http://codeforces.com/contest/1060/problem/E展开,给出一棵树,要计算对节点相连的两个节点再连边后所有点对的距离和。最初想用树上dp求解无果,后将问题转化,通过计算每条边的贡献,结合奇偶层距离计算,最终得出答案。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

http://codeforces.com/contest/1060/problem/E
题意:
给出一棵树,现对一个节点相连的两个节点再连一条边,问所有点对的距离和

自己做的时候还一直想着树上dp瞎搞,搞了半天弄不出来,发现这道题思路还是比较巧妙

思路:
先把问题转化一下,求树上所有点对的边距离和,那么针对每一条边他的贡献就是 一端点数*另一端点数 这是所有的要使用她的点对。 那么问题就被简化了,针对这道题目,可以得到距离为偶数的点 x/2,距离为奇数的为 (x+1)/2 ,那么就是奇偶层计算距离时需要加1,那么我们统计出这个再和之前每条边的贡献求和除2就是答案

#include<bits/stdc++.h>
#define fi first
#define se second
#define log2(a) log(n)/log(2)
#define show(a) cout<<a<<endl;
#define show2(a,b) cout<<a<<" "<<b<<endl;
#define show3(a,b,c) cout<<a<<" "<<b<<" "<<c<<endl;
#define tim printf("Time cost : %lf s\n",(double)clock()/CLOCKS_PER_SEC);
using namespace std;

typedef long long ll;
typedef long long LL;
typedef pair<int, int> P;
typedef pair<P, ll> LP;
const ll inf = 1e18;
const int N = 1e6 + 10;
const ll mod = 1e9 + 7;
const int base = 131;
const double pi = acos ( -1 );
const double eps = 1e-8;
inline ll ksm(ll a,ll b){ll ans=1;while(b){if(b&1)ans=ans*a%mod;a=a*a%mod,b>>=1;}return ans;}
#define a(i,j) a[(i-1)*m+(j)]
#define b(i,j) b[(i-1)*m+(j)]

ll  vis[N],num[N];
ll n, m,  k, x, y, z;
ll a[N], b[N],c[N];
ll cx, cy, cnt, ans, sum, flag, t, ff,q;

struct node
{
	ll to,k;
};
vector<int> v[N],res;

unordered_map<int, int> mp;
set<string> st;
ll dp[N];//该点下方的点数

void dfs(int x,int fa,int de)
{
	dp[x]=1;
	cnt+=de;
	for(int to:v[x])
	{
		if(to==fa) continue;
		dfs(to,x,de^1);
		dp[x]+=dp[to];
	}
}
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);

	cin>>n;

	for(int i=2;i<=n;i++)
	{
		cin>>x>>y;
		v[x].push_back(y);
		v[y].push_back(x);

	}
	dfs(1,-1,0);
	for(int i=1;i<=n;i++)
	{
		ans+=dp[i]*(n-dp[i]);
	}
	ans+=cnt*(n-cnt);//奇偶层的贡献
	cout<<ans/2;








}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值