【每日一题】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;
}
全部评论

相关推荐

一个菜鸡罢了:哥们,感觉你的简历还是有点问题的,我提几点建议,看看能不能提供一点帮助 1. ”新余学院“别加粗,课程不清楚是否有必要写,感觉版面不如拿来写一下做过的事情,教育经历是你的弱势就尽量少写 2. “干部及社团经历”和“自我评价”删掉 3. 论文后面的“录用”和“小修”啥的都删掉,默认全录用,问了再说,反正小修毕业前肯定能发出来 4. 工作经验和研究成果没有体现你的个人贡献,着重包装一下个人贡献
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务