BJ模拟：链上求和（树状数组）

题意：
给一棵树，求：

$$\sum_{x=1}^n \sum_{y=x+1}^{n} \sum_{k=1}^{dis(x,y)+1} ((x,y)路径上前k大点权和)$$
其中 $dis(x,y)$表示 $x$到$y$经过的边数。

题解：
一条路径上的一个点$u$会被统计比他小的点的个数+1次。 问题转化为：

$$\sum_{x=1}^n \sum_{y=x+1}^n \sum_u \sum_v a_u *[a_u > a_v] + \sum_{x=1}^n \sum_{y=x+1}^n \sum_u a_u$$
后面部分很好统计。
对于前面部分，考虑将$a_u$从小到大插入，用树状数组维护$dfs$序即可。 具体实现参见代码。

#include <bits/stdc++.h>
using namespace std;
typedef long long LL;

const int RLEN=1<<18|1;
inline char nc() {
    static char ibuf[RLEN],*ib,*ob;
    (ib==ob) && (ob=(ib=ibuf)+fread(ibuf,1,RLEN,stdin));
    return (ib==ob) ? -1 : *ib++;
}
inline int rd() {
    char ch=nc(); int i=0,f=1;
    while(!isdigit(ch)) {if(ch=='-')f=-1; ch=nc();}
    while(isdigit(ch)) {i=(i<<1)+(i<<3)+ch-'0'; ch=nc();}
    return i*f;
}

const int N=1e5+50, mod=1e9+7;
inline int add(int x,int y) {return (x+y>=mod) ? (x+y-mod) : (x+y);}
inline int dec(int x,int y) {return (x-y<0) ? (x-y+mod) : (x-y);}
inline int mul(int x,int y) {return (LL)x*y%mod;}

int n,ans,ind,a[N],fa[N],p[N],dfn[N],sze[N],bit1[N],bit2[N];
vector <int> edge[N];
inline void dfs(int x,int f) {
    dfn[x]=++ind; 
    sze[x]=1; fa[x]=f; 
    int rs=1, s=1;
    for(auto v:edge[x]) {
        if(v==f) continue;
        dfs(v,x); sze[x]+=sze[v];
        s=add(s,mul(rs,sze[v]));
        rs=add(rs,sze[v]);
    }
    s=add(s,mul(rs,n-sze[x]));
    ans=add(ans,mul(a[x],s));
}
inline void inc(int *bit,int p,int v) {for(int i=p;i<=n;i+=(i&(-i))) bit[i]=add(bit[i],v);}
inline int ask(int *bit,int p,int v=0) {for(int i=p;i;i-=(i&(-i))) v=add(v,bit[i]); return v;}
int main() {
    n=rd();
    for(int i=1;i<n;i++) {
        int x=rd(), y=rd();
        edge[x].push_back(y);
        edge[y].push_back(x);
    }
    for(int i=1;i<=n;i++) a[i]=rd(), p[i]=i;
    sort(p+1,p+n+1,[&](const int &i,const int &j) {return a[i]<a[j];});
    dfs(1,0);
    for(int i=1;i<=n;i++) {
        int x=p[i], v=a[x];
        int son_sze=ask(bit1,dfn[x]+sze[x]-1)-ask(bit1,dfn[x]-1);
        int par_sze=dec(ask(bit1,n),son_sze);
        par_sze=add(par_sze,ask(bit2,dfn[x]));
        ans=add(ans,mul(v,mul(par_sze,sze[x])));
        ans=dec(ans,v);

        inc(bit1,dfn[x],sze[x]);
        for(auto vt:edge[x]) {
            if(vt==fa[x]) continue;
            int t=dec(n,add(sze[vt],sze[x]));
            inc(bit2,dfn[vt],t);
            inc(bit2,dfn[vt]+sze[vt],mod-t);
        }
    } 
    memset(bit2,0,sizeof(int)*N);
    for(int i=n;i>=1;i--) {
        int x=p[i], v=a[x];
        int par_sze=ask(bit2,dfn[x]);
        ans=add(ans,mul(par_sze,sze[x]));
        for(auto vt:edge[x]) {
            if(vt==fa[x]) continue;
            int t=dec(n,sze[vt]);
            t=mul(t,v);
            inc(bit2,dfn[vt],t);
            inc(bit2,dfn[vt]+sze[vt],mod-t);
        }
    } 
    cout<<ans; 

}