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;}