0分求助,,搞不懂那里写丑了qwq
查看原帖
0分求助,,搞不懂那里写丑了qwq
584013
CY000楼主2022/3/3 11:30
#include <iostream>
#include <bits/stdc++.h>
using namespace std;
#define N 2000000+10
struct edge
{
    int dis,dis2,nex,to;
};
edge e[N*2];
int head[N],tot;
void add(int from,int to,int dis,int dis2)
{
    e[++tot].dis=dis;e[tot].dis2=dis2;
    e[tot].to=to;e[tot].nex=head[from];head[from]=tot;
}
int lg[N],fa[N][60],dep[N];
void DFS(int now,int fath)
{
    dep[now]=dep[fath]+1;
    fa[now][0]=fath;
    for(int i=1;i<=lg[dep[now]];i++)
    {
        fa[now][i]=fa[fa[now][i-1]][i-1];
    }
    for(int i=head[now];i;i=e[i].nex)
    {
        if(e[i].to!=fath)
            DFS(e[i].to,now);
    }
}
int LCA(int x,int y)
{
    if(dep[x]<dep[y])
    {
        swap(x,y);
    }
    while(dep[x]>dep[y])
    {
        x=fa[x][lg[dep[x]-dep[y]-1]];
    }
    if(x==y)
        return x;
    for(int k=lg[dep[x]]-1;k>=0;k--)
    {
        if(fa[x][k]!=fa[y][k])
        {
            x=fa[x][k];
            y=fa[y][k];
        }
    }
    return fa[x][0];
}
long long power[N];
long long ans;
void get(int now)
{
    int u;
    for(int i=head[now];i;i=e[i].nex)
    {
        if(e[i].to!=fa[now][0])
        {
            get(e[i].to);
            power[now]+=power[e[i].to];
        }
        else
            u=i;
    }
    if(e[u].dis*power[now]<e[u].dis2)
        ans+=e[u].dis*power[now];
    else
        ans+=e[u].dis2;
}
int main()
{
   // freopen("in.txt","r",stdin);
    int n;
    scanf("%d",&n);
    for(register int i=1,x,y,dis,dis2;i<=n-1;i++)
    {
        scanf("%d%d%d%d",&x,&y,&dis,&dis2);
        add(x,y,dis,dis2);
        add(y,x,dis,dis2);
    }
    for(register int i=1;i<=n;i++)
    {
        lg[i]=lg[i-1]+(1<<lg[i-1]==i);
    }
    DFS(1,0);
    for(register int i=1,a;i<=n-1;i++)
    {
        a=LCA(i,i+1);
        power[i]++,power[i+1]++,power[a]-=2;
    }
    get(1);
    cout<<ans<<endl;
    return 0;
}

2022/3/3 11:30
加载中...