第三场多校 03 HDU6058

Kanade's sum

这个题目主要的思路是对的

就是计算每个数字出现的次数

然后我发现我们可以对每个数字，往左找k个比他大的，往右找k个比他大的

然后他们的位置之差乘起来就是出现的次数

等下会补个图详细解释下

然而如果我们找k个比他大的话，线性找的话，时间复杂度会爆炸

所以我们就想从最小的开始找，然后找一个就把他从数组中删去

那么对于下一个数，他的左右k个就是k个比他大的数了

所以我们要保存每个数的位置

接下来的关键就是把一个数从数组中删去了

我们删去后，数组中他后面的所有数在数组中的位置都会改变

那么我们有要花费n的时间去修改位置，这是不划算的

现在题解就有一种方法，直接数组模拟链表

模拟++和--，真的是非常巧妙

#include <iostream>#include <cstdio>#include <cstring>#include <algorithm>#include <cmath>#define ll long long#define maxn 600000using namespace std;int n,kk;int a[maxn],pre[maxn],ne[maxn],poi[maxn];//pre[i]表示第i位的前面一位的位置ll cal(int k){    int pos=poi[k];//先找到k在数组中的位置    int q[100],h[100],nq(0),nh(0);    for (int i=pre[pos];i!=-1;i=pre[i])//往前找k个比它小的    {        nq++;q[nq]=poi[a[i]];        if (nq==kk) break;    }    if (nq<kk) q[++nq]=0;//如果不够k个就加上一个位置0，这个为什么画图就知道了，开始没注意，wa了好久    for (int i=ne[pos];i!=-1;i=ne[i])//同理    {        nh++;h[nh]=poi[a[i]];        if (nh==kk) break;    }    if (nh<kk) h[++nh]=n+1;    q[0]=pos;h[0]=pos;    ll ans(0);    for (int i=1;i<=nq;i++)//开始算    {        if (kk-i+1<=nh)            ans+=(ll)(q[i-1]-q[i])*(ll)(h[kk-i+1]-h[kk-i]);    }    int pp=pre[pos];//删去pos位    int nn=ne[pos];    if(pre[pos]!=-1)ne[pre[pos]]=nn;    if(ne[pos]!=-1)pre[ne[pos]]=pp;    pre[pos]=ne[pos]=0;    return ans;}int main(){    int t;    scanf("%d",&t);    while(t--)    {        scanf("%d %d",&n,&kk);        memset(pre,-1,sizeof(pre));        memset(ne,-1,sizeof(ne));        for (int k=1;k<=n;k++)        {            scanf("%d",&a[k]);            poi[a[k]]=k;            pre[k]=k-1;ne[k]=k+1;        }        pre[1]=-1;ne[n]=-1;        ll ans(0);        for (int k=1;k<=n-kk+1;k++)//从小到大计算        {            ans+=(ll)k*cal(k);            //cout<<zzz<<endl;        }        printf("%lld\n",ans);    }    return 0;}