【每日一题】Treepath

Treepath

http://www.nowcoder.com/questionTerminal/660faba350d74df095191ec3d321c15c

对于树上的点,只有深度为奇数的点走到深度为奇数的点和深度为偶数的点走到深度为偶数的点经过的边才会是偶数,dfs一边记录每个点的深度,统计深度为奇数的点的个数n和深度为偶数的点的个数m

ans = 图片说明

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
const int N = 1e5+10;
struct Node{
    int to,nex;
}e[N<<1];
int n,head[N],idx,depth[N];
int mx,cnt[N];//最大深度,深度为i的点有几个 
void add_edge(int u,int v){
    e[idx].to = v;
    e[idx].nex = head[u];
    head[u] = idx++;
}
void dfs(int u,int fa){
    depth[u] = depth[fa]+1;
    cnt[depth[u]]++;
    mx = max(mx,depth[u]); 
    for(int i = head[u];~i;i = e[i].nex){
        int v = e[i].to;
        if(v == fa) continue;
        dfs(v,u);
    }
}
int main(){
    scanf("%d",&n);
    memset(head,-1,sizeof(head));
    idx = 0;
    for(int i = 1;i < n;i++){
        int u,v;scanf("%d%d",&u,&v);
        add_edge(u,v);add_edge(v,u);
    }
    dfs(1,0);
    ll odd = 0,even = 0;//深度为奇数的点个数,深度为偶数的点的个数 
    for(int i = 1;i <= mx;i++){
        if(i&1) odd+=(1ll*cnt[i]);
        else even+=(1ll*cnt[i]);
    }
    ll res = odd*(odd-1)/2+even*(even-1)/2;
    printf("%lld\n",res);
    return 0;
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务