hiho Total Highway Distance

//dfs
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<vector>
#include<iostream>
using namespace std;
#define N 100010
int head[N];//head[i]代表起点是i的边的编号
int cnt;//代表边的编号
int n,m;
char opt[10];
long long ans=0;
int size[N],deep[N],dis[N];
struct edge
{
    int u;
    int v;
    int w;
    int next;//指向上一条边的编号
}e[2*N];
void addedge(int u,int v,int w)
{
    e[cnt].u=u;//边的起点
    e[cnt].v=v;//边的终点
    e[cnt].w=w;//边的权值
    e[cnt].next=head[u];
    head[u]=cnt++;
}
void solve()
{
    int u,v,w;
    memset(head,0xff,sizeof(head));
    cnt=0;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d%d",&u,&v,&w);
        addedge(u,v,w);
        addedge(v,u,w);
    }
}
//起点,深度,上一个节点
void dfs(int node,int de,int lastfa)
{
    deep[node]=de;
    size[node]=1;
    for(int i=head[node];i!=-1;i=e[i].next)
    {
        if(e[i].v!=lastfa)
        {
            dis[e[i].v]=e[i].w;//下面那个点记边的值
            dfs(e[i].v,de+1,node);
            ans+=(long long)size[e[i].v]*(n-size[e[i].v])*dis[e[i].v];
            size[node]+=size[e[i].v];//点的个数包括自己和下面的节点
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    solve();
    dfs(1,1,1);
    for(int i=1;i<=m;i++)
    {
        scanf("%s",opt);
        if(opt[0]=='Q')
            printf("%lld\n",ans);
        else
        {
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            if(deep[u]<deep[v])
                swap(u,v);
            ans+=(long long)size[u]*(n-size[u])*(w-dis[u]);
            dis[u]=w;
        }
    }
}

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值