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