Treepath

给定一棵n个点的树,问其中有多少条长度为偶数的路径。路径的长度为经过的边的条数。x到y与y到x被视为同一条路径。路径的起点与终点不能相同。
输入描述:
第一行一个数n表示点的个数;
接下来n-1行,每行两个整数x,y表示边;
保证输入数据形成一棵树;
1<=n<=100000
输出描述:
一行一个整数表示答案。

思路:题意:给出一棵树,求树上所有长度为偶数的路径个数
涉及知识点:树上dfs/dpdfs/dp
思路:以任意一个节点dfsdfs(默认以11号节点dfsdfs),因为树上任意两点之间的距离是固定的,所以我们可以dfsdfs得到所有距离11号节点的长度,存在两个结论(证明看下图):
①长度为偶数的任意两个节点之间的距离一定是偶数
②长度为奇数的任意两个节点之间的距离也一定是偶数
最后记录距离11号节点长度为奇数的节点个数cnt1,距离11号节点长度为偶数的节点个数cnt2
答案就是(cnt1*(cnt1-1)/2) + (cnt2*(cnt2-1)/2)

在这里插入图片描述

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn = 1e5 + 5;
struct node{
int t,nex;
};
node a[maxn<<1];
int head[maxn],tot,cnt[maxn];
void add(int x,int y){
  a[++tot].t = y,a[tot].nex = head[x],head[x] = tot;
}
void dfs(int x,int fa,int len)
{
  cnt[x] = len;
  for(int i=head[x]; i ; i=a[i].nex){
      if(a[i].t != fa)
          dfs(a[i].t , x , len + 1);
  }
}
int main()
{
  int n,x,y,cnt1 = 0 ,cnt2 = 0;
  cin>>n;
  for(int i=1;i<n;i++)
      cin>>x>>y,add(x,y),add(y,x);
  dfs(1 , 0 , 0);
  for(int i=1;i<=n;i++){
      if(cnt[i]%2)cnt1++;
      else cnt2++;
  }
  cout<<(1ll*cnt1*(cnt1 - 1)/2) + (1ll*cnt2*(cnt2 - 1)/2)<<endl;
  return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值