树上直径---巨木之森

定理---树上任意一点的最远点一定是树的直径的端点。

可以发现,要想遍历完整颗树,再回到原来的位置,那么就要把每条边都走两遍,但是现在不需要回到原来的位置,所以只要找到离根节点最远的点,用总权值的两倍减去 这条最远边的长度就好了。

于是,我们可以先以某点dfs,求出这点的最远距离----一定是树直径的是某个端点p。

再以该点p出发,dfs找出另一个端点 q;

最后,再dfs q点求出所有点到q点的距离,之后计算,排序求解即可。

#include<bits/stdc++.h>
//树的深搜 
using namespace std;
typedef long long ll;
const int N = 2e5 + 10, M = 2 * N;

ll n, m;
ll d[N];
bool st[N];
int h[N],e[M],ne[M],idx;
ll edge[N];
ll q[N];
ll dis1[N];
ll dis2[N];
void add (int a,int b,ll c)
{
    e[idx]=b;
    ne[idx]=h[a];
    edge[idx] = c;
    h[a]=idx++;
    
}

void dfs(ll dis[],int u,int fa)
{
    for(int i = h[u] ; i != - 1 ; i = ne[i] )
    {
        int j = e[i];
        if(j == fa) continue;
        dis[j] = max(dis[j],dis[u] +edge[i] );
        dfs(dis,j,u);
    }
}



int main()
{
    memset(h,-1,sizeof h);
    cin >> n >> m;
    ll x ,y,z;
    ll all = 0;
    for(int i = 1 ; i < n ;i++)
    {
        cin >> x >> y >> z;
        all += z;
        add(x,y,z);
        add(y,x,z);
        
    }
    int maxv = 1;
     dfs(dis1,maxv,0);
     for(int i = 2 ;i <= n ;i++) if(dis1[i] > dis1[maxv]) maxv = i;
     memset(dis1,0,sizeof dis1);
     dfs(dis1,maxv,0);
     int  se = 1;
     for(int i = 2 ;i <= n ;i++) if(dis1[i] > dis1[se]) se = i;
     dfs(dis2,se,0);
    //z = 2;
   //  cout << all << endl;
    for(int i = 1 ;i <= n ;i++)
    {
        q[i] = 2ll * all - max(dis1[i],dis2[i]);
    }
    sort(q + 1, q + 1 + n);
    ll ans = 0;
    int i = 1;
    for(;i<=n;i++)
    {
        if(q[i]>m)break;
        else m-=q[i];
    }
    cout << i - 1 << endl;
   // cout << i - 1 <<endl;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值