（ 求区间第k大及大于等于它的和 ）【 主席树扩展 】

（ 求区间第k大及大于等于它的和 ）【 主席树扩展 】

即求区间第1大+第2大+第3大+...+第k大的和。

比如：a = 1 2 3 4 5

query(1,5,5) = 5+4+3+2+1

query(2,4,2) = 4 + 3

query(1,3,1) = 3

引入sumt数组，表示当前点及其儿子节点的和。

int update( int pre, int left, int right, int x )
{
    int node = tot++;
    L[node] = L[pre];  // 必须要加
    R[node] = R[pre];
    tree[node] = tree[pre] + 1;
    sumt[node] = sumt[pre] + b[x];  // 新引入的的sumt
    if ( left<right ) {
        if ( x<=mid ) L[node] = update(L[pre],left,mid,x);
        else R[node] = update(R[pre],mid+1,right,x);
    }
    return node;
}

在查询时做一些修改，很容易理解

int query( int node1, int node2, int left, int right, int k )
{
    if ( left==right ) return b[left]*k; // 这个*k好好理解， b[]是离散化数组
    int rsum = tree[ R[node2] ] - tree[ R[node1] ];
    /// 找第k大及其更大的和
    if ( rsum>=k ) {
        return query( R[node1], R[node2], mid+1, right, k );
    }
    else {
        int sumr = sumt[ R[node2] ] - sumt[ R[node1] ]; // 右边的求和全部加上
        return sumr+query( L[node1],L[node2],left,mid,k-rsum );
    }
}

例题：Easy DP Problem

https://codeforces.com/gym/102770/problem/E

 

#include <bits/stdc++.h>
#define int long long
#define mid (left+right)/2

using namespace std;

const int maxn=1e5+10;
int n,m,q,tot=0;
int a[maxn],b[maxn],sum[maxn];
int T[maxn],tree[maxn*20],L[maxn*20],R[maxn*20];
int sumt[maxn*20];
// T[i]存的是第i棵树的root， tree[i]存的是正常线段树的值， l[i]存的是i号节点的左儿子

int built_tree( int left, int right )
{
    int node = tot++;
    if ( left<right ) {
        L[node] = built_tree(left,mid);
        R[node] = built_tree(mid+1,right);
    }
    return node;
}

int update( int pre, int left, int right, int x )
{
    int node = tot++;
    L[node] = L[pre];  // 必须要加
    R[node] = R[pre];
    tree[node] = tree[pre] + 1;
    sumt[node] = sumt[pre] + b[x];
    if ( left<right ) {
        if ( x<=mid ) L[node] = update(L[pre],left,mid,x);
        else R[node] = update(R[pre],mid+1,right,x);
    }
    return node;
}

// 求l~r，第1大，2大，3大...k大的和
int query( int node1, int node2, int left, int right, int k )
{
    if ( left==right ) return b[left]*k;
    int rsum = tree[ R[node2] ] - tree[ R[node1] ];
    // 找第k大及其更大的和
    if ( rsum>=k ) {
        return query( R[node1], R[node2], mid+1, right, k );
    }
    else {
        int sumr = sumt[ R[node2] ] - sumt[ R[node1] ];
        return sumr+query( L[node1],L[node2],left,mid,k-rsum );
    }
}

signed main()
{
    sum[0] = 0;
    for ( int i=1; i<=100000; i++ ) sum[i]=i*i+sum[i-1];
    int listt;
    cin >> listt;
    while ( listt-- ) {
        tot = 0; // 主席树动态开点，从0开始
        memset(T,0,sizeof(T));memset(tree,0,sizeof(tree));
        memset(L,0,sizeof(L));memset(R,0,sizeof(R));
        cin >> n ;
        for ( int i=1; i<=n; i++ ) {
            scanf("%lld",&a[i]); b[i]=a[i];
        }
        sort(b+1,b+1+n);
        m = unique(b+1,b+1+n) - b - 1; // 离散化
        T[0] = built_tree(1,m);
        for ( int i=1; i<=n; i++ ) {
            a[i] = lower_bound(b+1,b+1+m,a[i]) - b;
            T[i] = update(T[i-1],1,m,a[i]);
        }
        cin>>q;
        while ( q-- ) {
            int l,r,k;
            scanf("%lld %lld %lld",&l,&r,&k);
            printf("%lld\n",query(T[l-1],T[r],1,m,k)+sum[r-l+1]);
        }
    }

    return 0;
}
/*
1
5
1 2 3 4 5
3
1 5 5
1 3 2
3 3 1
l r k 求l~r，第1大，2大，3大...k大的和

*/