题意
给出一个n和一个k,求1~n的每个区间的第k大的数的总和是多少,区间长度小于k的话,贡献为0。
题解
因为给出的n个数是1-n的全排列,所以我们从小开始枚举第k大的值,即枚举第k大的值从1到n-k+1。剩下的值不可能有k-1个数比它大所以不可能是第k大数,贡献为0,所以不用枚举。对于每个枚举的值,从该值的位置向前找k个比它大的数的位置,向后找k个比它大的数的位置。然后就可以计算所有第k大数为枚举值的区间个数。枚举值和区间个数相乘,对于每个枚举值最后再相加即为最后答案。
比如我们令k=3,枚举值为5,那么向前和向后比它大的3个数的位置之间位置关系如下:
pre[3]–pre[2]–pre[1]–5=pre[0]=pos[0]–pos[1]–pos[2]–pos[3]
所以我们枚举区间右边界i,此时满足区间第3大数为5的区间个数为(pre[i]-pre[i+1])*(pos[k-i]-pos[k-i-1])(你可以把i=0带入一下就知道了)
可以结合代码看看
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 5e5+5;
//a[]保存n个数的值,loc[]保存每个值的位置
int a[maxn],loc[maxn],n,k;
//pre[i]保存前面比位置为i的数大的数的位置
//pos[i]保存后面比位置为i的数大的数的位置
//pr[i]保存前面第i个比枚举值大的数的位置
//po[i]保存后面第i个比枚举值大的数的位置
//pr_num表示前面比枚举值大的数的个数
//po_num表示后面比枚举值大的数的个数
int pre[maxn],pos[maxn],pr[maxn],po[maxn],pr_num,po_num;
void erase(int x) //删除操作
{
int pp=pre[x];
int nn=pos[x];
if(pre[x]) pos[pp]=nn;
if(pos[x]<=n) pre[nn]=pp;
pre[x]=pos[x]=0;
}
int main()
{
//freopen("in.txt","r",stdin);
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
loc[a[i]]=i;
}
for(int i=1;i<=n;i++) pre[i] = i-1,pos[i] = i+1;
ll ans=0;
for(int i=1;i<=n-k+1;i++)
{
int p = loc[i];
pr_num = po_num = 0;
for(int x=p;x>=1&&pr_num<=k;x=pre[x]) pr[pr_num++] = x;
for(int x=p;x<=n&&po_num<=k;x=pos[x]) po[po_num++] = x;
pr[pr_num++] = 0;
po[po_num++] = n+1;
for(int j=0;j<pr_num-1;j++)
{
if(k-j>0&&k-j<po_num)
{
ans += 1ll*(pr[j]-pr[j+1])*(po[k-j]-po[k-j-1])*i;
}
}
//这里必须要删除才能保证对枚举的每个数pre和pos都是比它大的数的位置
erase(p);
}
printf("%lld\n",ans);
}
//printf("Time used = %.2f\n",(double)clock() / CLOCKS_PER_SEC);
return 0;
}
我可能讲的还不是能一下让人理解,但是你看完题解再多看看几遍代码,最好自己画一下(可以就用样例),还是能懂的。这里其实就相当于用数组实现一个链表维护从小开始的数的前面和后面比它大的数。