题意:
在 一个数列中,对于所有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;
}