hdu 6058

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=6058

思路：找每个序列的第K大的数的和，其实我们可以枚举a[i]左右两边比它大的各K个数，然后使得a[i]在这个序列保持第k大，这样我们就可以直接a[i]*left*right每次都是a[i]为k大然后通过不断移动使得这个成立，这样就可以枚举完这个数，其他数也按照这方法进行。

代码;

 

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int N=500000+10;
int l[N],r[N],a[N];
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        int n,k;     int i,j;
        scanf("%d %d",&n,&k);
        for(i=0;i<n;i++)
        {
            scanf("%d",&a[i]);
        }

        memset(l,0,sizeof(l));
        memset(r,0,sizeof(r));
        long long int ans=0;
        for(i=0;i<n;i++)
        {
            int left=1,right=1;
            for(j=i+1;j<n;j++)
            {
                if(right>k)
                {
                    break;
                }
                if(a[j]>a[i])
                {
                    r[right++]=j-i;
                }
            }
            if(j>=n)
                r[right]=n-i;
            for(j=i-1;j>=0;j--)
            {
                if(left>k)
                {
                    break;
                }
                if(a[j]>a[i])
                {
                    l[left++]=i-j;
                }
            }
            if(j<=0)
                l[left]=i+1;
            for(j=0;j<left;j++)
            {
                if(k-j-1>=right)
                    continue;
                int x=l[j+1]-l[j];
                int y=r[k-j]-r[k-j-1];
                ans+=(long long)a[i]*x*y;
            }
        }
        printf("%lld\n",ans);

    }
    return 0;
}