美团 18 Round1 C-城市漫游 思维+LCA

题意:n个节点的树,每条边有权值w[i]和至少要经过的次数L[i].
n,Q<=1e5,w[i],1<=l[i],w[i]<=1e9 . Q次询问,问从s->t 满足所有边的L值时,需要的最小花费.


s-t路径上的边其l[i]值为奇数 则显然正好可以满足.若为偶数 则需要多走一次.
除了s-t路径上的边,显然都要从从(s,t)出发然后在回到(s,t).于是其l[i]值为偶数的可以正好满足,奇数的需要多走一次.


只要先记录所有边经过偶数次的和tot.然后要扣掉一次(s,t)路径中l[i]为奇数的边权.加上一次(s,t)中l[i]偶数的边权即可.

建树时,只要令l[i]为奇数的边权为-w,l[i]偶数的边权为w .LCA配合查询即可.

注意负边要先加mod在取余

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> ii;
const int N=2e5+5,M=23,mod=1e9+7;
vector<ii> e[N];
ll n,fa[N][M],dis[N],dep[N];
void dfs(int u,int par){
    fa[u][0]=par;
    dep[u]=dep[par]+1;
    for(int i=1;i<M;i++) fa[u][i]=fa[fa[u][i-1]][i-1];
    for(int i=0;i<e[u].size();i++)
    {
        ll v=e[u][i].first,w=e[u][i].second;
        if(v==par)  continue;
        dis[v]=(dis[u]+w)%mod;
        dfs(v,u);
    }
}
int LCA(int u,int v){
    if(dep[u]>dep[v])    swap(u,v);
    if(dep[u]<dep[v]){
        int del=dep[v]-dep[u];
        for(int i=0;i<M;i++)if((del>>i)&1) v=fa[v][i];
    }
    if(u!=v){
        for(int i=M-1;i>=0;i--){
            if(fa[u][i]!=fa[v][i])
                u=fa[u][i],v=fa[v][i];
        }
        u=fa[u][0],v=fa[v][0];
    }
    return u;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cin>>n;
    ll tot=0,Q,u,v;
    for(int i=1;i<=n-1;i++)
    {  
        ll w,l;
        cin>>u>>v>>w>>l;
        if(l%2) tot=(tot+((l+1)*w)%mod)%mod,w=(-w+mod)%mod;
        else    tot=(tot+(l*w)%mod)%mod;
        e[u].push_back(ii(v,w));
        e[v].push_back(ii(u,w));
    }
    dfs(1,0);
    cin>>Q;
    while(Q--){
        cin>>u>>v;
        ll x=LCA(u,v),d=(dis[u]+dis[v])%mod;
        d=(d-dis[x]+mod)%mod;
        d=(d-dis[x]+mod)%mod;
        ll res=(d+tot)%mod;
        cout<<res<<'\n';
    }  
    return 0;
}