HDU 6058 枚举

枚举

题意：

有一列数字，现在问所有的区间的第k大的数字之和是多少？

思路：

直接去把每一个区间求出第k大的数字求出会超时，但是分析k值可以发现范围特别小，所以可以选择枚举每一个数字，把枚举的数字当做第k大值，求出其贡献值即可。

为了求出第k大值其区间范围只能是比其大的k 个数字，分为左右两边，枚举每一次都求出左右的两个区间的比这个数字大的k个数字，找出之后合法区间为：若这个数字左边有x 个比这个数字大的数字那么右边就有k-x-1个数字。枚举每一个区间即可，其中穿插了一个组合数：比如 6 2 1 4 3 8 7 9 当3作为枚举的数字的时候k = 2其 左边为6 4 当左边有1个比3打的时候其实有三种情况分别是2 1 4，右区间类似。

#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

const int maxn = 1000005;

int n,k;
int a[maxn];
int l[maxn],r[maxn],lnum,rnum;

int main(int argc, char const *argv[])
{
    //freopen("in.txt","r",stdin);

    int ncase;
    scanf("%d",&ncase);
    while(ncase--) {
        memset(a,0,sizeof(a));
        memset(l,0,sizeof(l));
        memset(r,0,sizeof(r));
        scanf("%d%d",&n,&k);
        for(int i = 0;i < n; i++) {
            scanf("%d",&a[i]);
        }
        long long ans = 0;
        for(int i = 0;i < n; i++) {
            lnum = rnum = 1;
            int j;
            for(j = i + 1;j < n; j++) {
                if(rnum > k)
                    break;
                if(a[j] > a[i]) {
                    r[rnum++] = j-i;
                }
            }
            if(j >= n) {
                r[rnum] = n - i;
            }
            for(j = i - 1;j >= 0; j--) {
                if(lnum > k)
                    break;
                if(a[j] > a[i])
                    l[lnum++] = i-j;
            }
            if(j <= 0)
                l[lnum] = i+1;
            for(j = 0;j < lnum; j++) {
                if(k - j - 1 >= rnum)
                    continue;
                int cnt1 = l[j+1] - l[j];
                int cnt2 = r[k-j] - r[k-j-1];
                ans += (long long)a[i]*cnt1*cnt2;
            }
        }
        printf("%I64d\n",ans);
    }

    return 0;
}