2017 多校3 1003 Kanade's sum

http://acm.hdu.edu.cn/showproblem.php?pid=6058

这道题,主要是反过来想,求一段序列的所有非空子序列中第k大的数的和

可以理解为是求每一个数在多少个非空子序列中担任第k大的位置

所以要从当前数的前面和后面分别找出k-1个别当前数大的数

那么就可以从前到后枚举有多少个满足当前数为第k大的序列了

注意 因为是找第k大,所以比当前数小的数可以不用考虑了.

#include<bits/stdc++.h>
using namespace std;
int a[666666];
int rig[666666];
int lef[666666];
int main(){
    int t;
    cin>>t;
    while(t--)
    {
        int n,k;
        scanf("%d%d",&n,&k);
        int i,j;
        for(i=0;i<n;i++)
        scanf("%d",&a[i]);
        long long int ans=0;
        for(i=0;i<n;i++)
        {
            int le=1,ri=1;
            for(j=i+1;j<n;j++)
            {
                if(ri>k)break;
                if(a[j]>a[i])
                {
                    rig[ri++]=j-i;
                }
            }
            if(j>=n)
            {
                rig[ri]=n-i;
            }
            for(j=i-1;j>=0;j--)
            {
                if(le>k)break;
                if(a[j]>a[i])
                {
                    lef[le++]=i-j;
                }
            }
            if(j<=0)
            {
                lef[le]=i+1;
            }
            for(j=0;j<le;j++)
            {
                if(k-j-1>=ri)continue;
                int aa=lef[j+1]-lef[j];
                int bb=rig[k-j]-rig[k-j-1];
                ans+=(long long int )(aa*bb)*a[i];
            }
        }
        cout<<ans<<endl;
    }
    return 0;
}