Delivery Service(tarjan lca)_deliveryservice-CSDN博客

本文链接：https://blog.csdn.net/m0_37802215/article/details/81604192

一棵树,2e5个点,2e5个点对,要你对这些点对经过的边进行统计,即每条边被经过的次数进行统计.
进行两次dfs,一次tarjan,找到所有点对的lca,然后dfs是维护每个点上一个num[p],代表这个子树中仍然在等待另一半的点数,那么这个节点的父节点在像这个子树连边的时候,这条边的访问次数就要+num[p].
这个num[p]怎么维护呢,首先他会等于所有子树的num[p]之和,包括他自己身上有的点对,也要num[p]++.
然后假如他是一个点对的lca,意味着他的子树中已经有了两个会”互相抵消”的,
所以他的num[p]-=2*lca[p].
注意在tarjan时,如果一个点的对象的父节点就是自己,代表他自己就是那个lca,这里我们不再加一.

#include <iostream>
#include <vector>
#include <set>
#include <algorithm>
#define debug(x) std::cerr << #x << " = " << (x) << std::endl
using namespace std;
typedef long long LL;
const int MAXN = 5e5 + 17;
const int MOD = 998244353;
vector<pair<int,int > > G[MAXN];
int num[MAXN],tms[MAXN],vis[MAXN],cut[MAXN];
vector<int > fate[MAXN];
int par[MAXN],rk[MAXN];
int fd(int x) 
{
    return par[x]==x?x:par[x]=fd(par[x]);
}
void unite(int x,int y)
{
    int a = fd(x);
    int b = fd(y);
    if(a==b) return;
    if(rk[a]>rk[b])
        par[b] = a;
    else
        par[a] = b;
    if(rk[a]==rk[b])
        rk[a]++;
}
void tarjan(int p)
{
    vis[p] = 1;
    for(auto i:G[p])
    {
        int v = i.first,id = i.second;
        if(!vis[v])
        {
            tarjan(v);
            par[v] = p;
        }
    }
    for(auto i: fate[p])
        if(vis[i]&&fd(i)!=p)
            cut[fd(i)]++;
}
void dfs(int p,int f)
{
    vis[p] = 1;
    for(auto i: G[p])
    {
        int v = i.first,id = i.second;
        if(!vis[v])
        {
            dfs(v,p);
            tms[id] += num[v];
            num[p] += num[v];
        }
    }
    for(auto i: fate[p])
        num[p]++;
    num[p] -= 2*cut[p];
}
void init(int n)
{
    for (int i = 0; i < n; ++i)
        par[i] = i;
}
int main()
{
#ifdef noob
    freopen("Input.txt", "r", stdin);
    freopen("Output.txt", "w", stdout);
#endif
    int n;
    cin>>n;
    init(n);
    vector<int > cost;
    for (int i = 0; i < n-1; ++i)
    {
        int u,v,c;
        scanf("%d%d%d",&u,&v,&c);
        u--,v--;
        G[u].push_back({v,i});
        G[v].push_back({u,i});
        cost.push_back(c);
    }
    int q,k = 0;
    cin>>q;
    while(q--)
    {
        int u,v;
        scanf("%d%d",&u,&v);
        u--,v--;
        fate[u].push_back(v);
        fate[v].push_back(u);
    }
    tarjan(0);
    for (int i = 0; i < n; ++i)
        vis[i] = 0;
    dfs(0,-1);
    vector<int > ord;
    for (int i = 0; i < n-1; ++i)
        ord.push_back(tms[i]);
    sort(cost.begin(), cost.end());
    sort(ord.begin(), ord.end());
    LL ans = 0;
    for (int i = n-2,j = 0; i > -1; --i,++j)
        ans += 1LL*ord[i]*cost[j];
    cout<<ans<<endl;
    return 0;
}