权值线段树学习笔记

权值线段树所存的值并非是下标，而是数组中数的值，可以很方便的解决诸如求区间第k大数的问题

记录一下

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int w[N],a[N];
int n;
struct node{
    int l,r,sum;
}tr[N*8];
void build(int root,int l,int r){
    tr[root]={l,r,0};
    if(l == r) return;
    int mid = l + r >> 1;
    build(root<<1,l,mid);
    build(root<<1|1,mid+1,r);
}
void pushup(int root){
    tr[root].sum = tr[root<<1].sum + tr[root<<1|1].sum;
}
void change(int root,int pos,int val){
    if(tr[root].l == pos && tr[root].r == pos){
        tr[root].sum += val;
        return;
    }
    int mid = tr[root].l + tr[root].r >> 1;
    if(mid>=pos) change(root<<1,pos,val);
    else change(root<<1|1,pos,val);
    pushup(root);
}
int query(int root,int l,int r){
    if(l > r) return 0;
    if(tr[root].l == l && tr[root].r == r) return tr[root].sum;
    int mid  = tr[root].l + tr[root].r >> 1;
    if(mid<l) return query(root<<1|1,l,r);
    else{
        if(r<=mid) return query(root<<1,l,r);
        else return query(root<<1,l,mid) + query(root<<1|1,mid+1,r);
    }
}
int main(){
    cin>>n;
    for(int i=1;i<=n;i++) cin>>w[i],a[i] = w[i];
    sort(w+1,w+n+1);
    for(int i=1;i<=n;i++){
        a[i] = lower_bound(w+1,w+n+1,a[i]) - w;
    }
    long long ans = 0;
    build(1,1,n+1);
    for(int i=1;i<=n;i++){
        ans+=query(1,a[i]+1,n+1);
        change(1,a[i],1);
    }
    cout<<ans<<endl;
    return 0;
}

注意查找操作

也可以参考一下牛客网的讲解
牛客网