hdu 6058 Kanade's sum

来源:互联网 发布:linux apache2.4 php7 编辑:程序博客网 时间:2024/05/24 07:09

题意:
在 一个数列中,对于所有l,r, 求和【l,r】区间内第k大的数

思路:

比赛的时候,队友已经想到对于每个数找这个数在多少个区间是第k大的,满足的区间个数乘以此数的和求和就是答案

考虑时间复杂度,遍历n个数,求区间数不能是n的复杂度,然后就想到了set优化,写完以后就是一直T,......

赛后看见那些50行左右的暴力代码过了。。。。。。

看见大佬们的暴力代码还是很精简的。。。。。。

必须承认,代码能力强的时候,暴力也能过。。。。。

#include<stdio.h>int a[500009];int l[100],r[100];int main(){    int t;    int n,k;    int li,ri,j;    long long ans=0;    scanf("%d",&t);    while(t--)    {        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++)        {            li=1;            ri=1;            for(j=i+1;j<n;j++)            {                if(ri>k) break;                if(a[j]>a[i]) r[ri++]=j-i;            }            if(j>=n) r[ri]=n-i;            for(j=i-1;j>=0;j--)            {                 if(li>k) break;                 if(a[j]>a[i]) l[li++]=i-j;            }            if(j<=0) l[li]=i+1;            int ll,rr;            for(j=0;j<li;j++)            {                if(k-j-1>=ri) continue;                ll=l[j+1] - l[j];                rr=r[k-j] - r[k-j-1];                ans+=(long long)a[i]*ll*rr;            }         }printf("%lld\n",ans);    }    return 0; } 



原创粉丝点击