2019ICPC南昌邀请赛网络赛J题 树上主席树&链上边权第k大

裸的树上主席树

需要注意：

 1.最后需要查询的部分是：query(1,n,root[l],root[r],root[lca],root[lca],k))

即l+r的前缀和减去两个lca的前缀和

2.这里的倍增操作，需要使用log数组先预处理出来，否则会超时

#include<bits/stdc++.h>
using namespace std;                                                                                                                                                                                                                                                                               using namespace std;
const int maxx=1e5+6;
struct node{
    int l,r,sum;
    }t[maxx*50];
struct edge{
    int to,d;
};
vector<edge>e[maxx];vector<int>rk;
int dep[maxx],f[maxx][25],lg[maxx],root[maxx],vis[maxx],cnt=0,n,m;
int getid(int x){
    return lower_bound(rk.begin(),rk.end(),x)-rk.begin()+1;
}
int LCA(int u,int v){
    if(dep[u]<dep[v]){
        swap(u,v);
    }
    while(dep[u]!=dep[v]){
        int d=dep[u]-dep[v];
//        cout<<d<<endl;
        if(lg[d]-1>=0)u=f[u][lg[d]-1];
        else u=f[u][0];
//        cout<<v<<"qwe"<<endl;
    }
//    cout<<"erte"<<endl;
    if(u==v)return u;
    for(int i=lg[dep[v]];i>=0;i--){
        if(f[u][i]!=f[v][i]){
            u=f[u][i];
            v=f[v][i];
        }
    }
    return f[u][0];
}
void update(int l,int r,int &cur,int pre,int k){
    t[++cnt]=t[pre];
    t[cnt].sum++;
    cur=cnt;
    if(l==r)return;
    int mid=(l+r)>>1;
    if(k<=mid)update(l,mid,t[cur].l,t[pre].l,k);
    else update(mid+1,r,t[cur].r,t[pre].r,k);
}
void dfs(int x,int fa,int d){
    vis[x]=1;dep[x]=d;
    for(int i=0;i<e[x].size();i++){
        int j=e[x][i].to;
        if(!vis[j]){
            f[j][0]=x;
            update(1,n,root[j],root[x],getid(e[x][i].d));
            dfs(j,x,d+1);
        }
    }
}
int query(int l,int r,int rt1,int rt2,int rt3,int rt4,int k){
    if(r<=k) return t[rt1].sum+t[rt2].sum-t[rt3].sum-t[rt4].sum;
    if(k<l)return 0;
    int mid=(l+r)>>1;
    if(k<=mid) return query(l,mid,t[rt1].l,t[rt2].l,t[rt3].l,t[rt4].l,k);
    else return (t[t[rt1].l].sum+t[t[rt2].l].sum-t[t[rt3].l].sum-t[t[rt4].l].sum)+query(mid+1,r,t[rt1].r,t[rt2].r,t[rt3].r,t[rt4].r,k);
}
int main(){
//    ios::sync_with_stdio(0);
//    cin.tie(0);cout.tie(0);
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)lg[i]=lg[i-1]+(1<<lg[i-1]==i);
    for(int i=1;i<n;i++){
        int u,v,d;
//        cin>>u>>v>>d;
        scanf("%d%d%d",&u,&v,&d);
        e[u].push_back({v,d});
        e[v].push_back({u,d});
        rk.push_back(d);
    }
    sort(rk.begin(),rk.end());
    rk.erase(unique(rk.begin(),rk.end()),rk.end());
//    for(int i=0;i<rk.size();i++)cout<<rk[i]<<endl;
    dfs(1,0,1);
//    dfs(1,0);
//    cout<<"asd"<<endl;
    for(int j=1;j<=20;j++){
        for(int i=1;i<=n;i++){
            f[i][j]=f[f[i][j-1]][j-1];
        }
    }
    for(int i=1;i<=m;i++){
        int l,r,k;
        scanf("%d%d%d",&l,&r,&k);
        int lca=LCA(l,r);
//        cout<<"zxfc"<<endl;
        k=upper_bound(rk.begin(),rk.end(),k)-rk.begin();
        if(k)printf("%d\n",query(1,n,root[l],root[r],root[lca],root[lca],k));
        else puts("0");
    }
}