E - RXD and dividing HDU - 6060 多校第三场

第三场。。大量的数学题。。最咸鱼的一次。。纠结于1003却忽视了这道水题;
就是题目难理解一点
求出所有子树的大小与k比较,然后用小的乘上权值求和就行

#include<bits/stdc++.h>  
using namespace std;  
struct node
{  
    int v,w;  
};  

node temp;  
vector<node> vec[1000005];  
int Size[1000005];  
int w[1000005];  

void dfs(int u,int pre){  
    Size[u]=1;  
    int len=vec[u].size();  
    for(int i=0;i<len;i++){  
        int v=vec[u][i].v;  
        if(v!=pre){  
            w[v]=vec[u][i].w;
            dfs(v,u);  
            Size[u]+=Size[v];  
        }  
    }  
}  

int main(){  
    int n,k;  
    while(~scanf("%d%d",&n,&k)){  
        for(int i=1;i<=n;i++){  
            vec[i].clear();  
            Size[i]=0;  
            w[i]=0;  
        }  
        for(int i=1;i<=n-1;i++){  
            int u,v,w;  
            scanf("%d%d%d",&u,&v,&w);  
            temp.v=v;  
            temp.w=w;  
            vec[u].push_back(temp);  
            temp.v=u;  
            vec[v].push_back(temp);  
        }  
        dfs(1,0);  
        long long sum=0;  
        for(int i=2;i<=n;i++){
            sum+=(long long)w[i]*min(Size[i],k); 
        }  
        printf("%lld\n",sum);  
    }  
    return 0;  
} 
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值