HDU 6058 Kanade's sum

题目来源
Kanade’s sum
Problem Description
Give you an array A[1..n]of length n.

Let f(l,r,k) be the k-th largest element of A[l..r].

Specially , f(l,r,k)=0 if r−l+1

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
const int maxn=500000+50;//数组个数
int a[maxn];
int b[255];//用于存放比当前数大的周围数字，如果可能的话左右各找k个就够了
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        int n,k;
        long long int ans=0;//用于存放最终结果
        scanf("%d%d",&n,&k);
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        for(int i=0;i<n;i++)
        {
            long long int res=0;//用于存放以当前点为中心的满足条件的排列个数
            int bl=0,br=0;//bl表示向左找到大于a[i]的个数,br表示向右找到大于a[i]的个数
            for(int j=i-1;j>=0;j--)
            {//向左遍历，找到k个大于a[i]的数之后就结束或j==0
                if(a[j]>a[i])
                {
                    bl++;
                    b[k-bl]=j;
                }
                if(bl==k)
                    break;
            }
            for(int j=i+1;j<n;j++)
            {//向右遍历，找到k个大于a[i]的数之后就结束或j==n
                if(a[j]>a[i])
                {
                    br++;
                    b[k+br]=j;
                }
                if(br==k)
                    break;
            }
            b[k]=i;
            if(br<k)
            {//如若右边找到的数小于k个
                br++;
                b[k+br]=n;
            }
            if(k-bl!=0&&k-bl+k<=k+br)//如若左边找到的数小于k个
                res=res+(fabs)(b[k-bl]+1)*(fabs)(b[k-bl+k]-b[k-bl+k-1]);
            //for(int j=k-bl;j<=k+br;j++)
              //  printf("**%d,%d ",j,b[j]);
            //printf("\n");
            for(int j=k-bl+1;j<=k+br;j++)
            {
                if(j+k>k+br)
                    break;//数已超出数组范围
                res=res+(b[j]-b[j-1])*(b[j+k]-b[j+k-1]);
                //printf("**&&res:%d\n",res);
            }
            ans=(long long)(ans+a[i]*res);
            //printf("ans:%d %lld\n",a[i],ans);
        }
        printf("%lld\n",ans);
    }
    return 0;
}