每日一题 P6869 [COCI2019-2020#5] Putovanje 树上差分

#include <bits/stdc++.h>
#define MAXN 500005
#define int long long
using namespace std;
struct EDGE
{
    int to,next,c1,c2;
} edge[MAXN];
int head[MAXN],ptr;
void add_edge(int u,int v,int c1,int c2)
{
    edge[++ptr].to=v;
    edge[ptr].next=head[u];
    edge[ptr].c1=c1;
    edge[ptr].c2=c2;
    head[u]=ptr;
}
void add(int u,int v,int c1,int c2)
{
    add_edge(u,v,c1,c2);
    add_edge(v,u,c1,c2);
}
int dep[MAXN],f[MAXN][21],cnt[MAXN];
int n,k;
void dfs(int now,int fa)
{
    for(int p=head[now]; p; p=edge[p].next)
    {
        int to=edge[p].to;
        if(to==fa) continue;
        dep[to]=dep[now]+1;//计算深度
        f[to][0]=now;
        dfs(to,now);
    }
}
void dp()
{
//预处理 f[u][i] u结点往上走2^i次
    for(int i=1; (1<<i)<=n; i++)
        for(int u=1; u<=n; u++)
            f[u][i]=f[f[u][i-1]][i-1];//u结点往上走2^i次等于网上走两个2^(i-1)
}
int lca(int x,int y)
{
    int p,t;
    if(dep[x]<dep[y])
        swap(x,y);//保证x比y大
    for(p=0; (1<<p)<=dep[x]; p++);//算出dp最多往上走2的多少次;
    for(t=--p; t>=0; t--)//从x走到y同一层
        if(dep[x]-(1<<t)>=dep[y])//不超过的话就往上走
            x=f[x][t];
    if(x==y) return x;
    for(t=p; t>=0; t--)//x和y一起走 不相等就往上走
        if(f[x][t]!=f[y][t])
            x=f[x][t],y=f[y][t];
    return f[x][0];
}
int ans=0;
void dfs2(int now,int fa,int fe)
{
    for(int p=head[now];p;p=edge[p].next)
    {
        int to=edge[p].to;
        if(to==fa)continue;
            dfs2(to,now,p);
    }
    cnt[fa]+=cnt[now];
    ans+=min(cnt[now]*edge[fe].c1,edge[fe].c2);
}
signed main()
{
    cin>>n;
    for(int i=1; i<n; i++)
    {
        int a,b,c1,c2;
        cin>>a>>b>>c1>>c2;
        add(a,b,c1,c2);
    }
    dfs(1,0);dp();
    for(int i=2;i<=n;i++)
    {
        cnt[lca(i-1,i)]-=2;
        cnt[i-1]++,cnt[i]++;
    }
    dfs2(1,0,0);
    cout<<ans;
}