Codeforces Round #364 (Div. 2) E. Connecting Universities【两次dfs】

两次dfs,从1开始搜索,son[x]记为以x为根的子树里有多少所大学,然后找到第一个son[x]>=m的做为下一次dfs的起点。第二次dfs以x为起点,计算到每一所大学的距离,然后在累加一下。

/* ***********************************************
Author        :Maltub
Email         :xiang578@foxmail.com
Blog          :htttp://www.xiang578.com
************************************************ */

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>
#include <queue>
#include <set>
#include <map>
#include <string>
#include <math.h>
#include <stdlib.h>
#include <time.h>
//#include <bits/stdc++.h>
#define rep(i,a,n) for(int i=a;i<n;i++)
#define per(i,a,n) for(int i=n-1;i>=a;i--)
#define pb push_back
using namespace std;
typedef vector<int> VI;
typedef long long ll;
const ll mod=1000000007;
const int N=2*1e5+10;
int u[N],dp[N],son[N],st,n,m;
vector<int>g[N];

void dfs(int x,int fa)
{
    if(u[x]) son[x]=1;
    else son[x]=0;
    for(int i=0; i<g[x].size(); i++)
    {
        int y=g[x][i];
        if(y==fa) continue;
        dfs(y,x);
        son[x]+=son[y];
    }
    //printf("%d %d\n",x,son[x]);
    if(son[x]>=m&&st==-1)
    {
        st=x;
    }
}

void dfs2(int x,int fa,int cnt)
{
    dp[x]=cnt;
    for(int i=0; i<g[x].size(); i++)
    {
        int y=g[x][i];
        if(y==fa) continue;
        dfs2(y,x,cnt+1);
    }
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("out.txt","w",stdout);
    int t;
    scanf("%d%d",&n,&m);
    memset(u,0,sizeof(u));
    memset(dp,0,sizeof(dp));
    for(int i=1; i<=2*m; i++)
    {
        scanf("%d",&t);
        u[t]=1;
    }
    for(int i=1; i<n; i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        g[a].push_back(b);
        g[b].push_back(a);
    }
    st=-1;
    dfs(1,0);
    //printf("%d\n",st);
    dfs2(st,0,0);
    ll ans=0;
    for(int i=1; i<=n; i++)
    {
        if(u[i]) ans+=dp[i];
    }
    cout<<ans<<endl;
    return 0;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值