HDU 6058 Kanade's sum（链表）

#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
/*
链表
给定一个长度为n的数列，求所有区间第k大的数之和
*/
const int maxn=5e5+5;
int n,k;
int a[maxn],pos[maxn];
int pre[maxn],nex[maxn];


bool cmp(int x,int y)
{
    return a[x]<a[y];
}

int main()
{
    int T;
    int kase=1;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
        }
        for(int i=0;i<=n+1;i++)
        {
            pre[i]=i-1;
            nex[i]=i+1;
            pos[i]=i;
        }
        sort(pos+1,pos+1+n,cmp);
        LL ans=0;
        for(int i=1;i<=n;i++)
        {
            int pl=pos[i];
            int pr=pos[i];
            int cnt=0;
            //直接找k-1个数，这k-1个数都是比当前遍历的这个数大，
            //因为通过不断维护链表，保证当前遍历的数是链中最小的数
            while(cnt<k-1&&pre[pl]>0)
            {
                cnt++;
                pl=pre[pl];
            }
            while(cnt<k-1&&nex[pr]<=n)
            {
                cnt++;
                pr=nex[pr];
            }

            if(cnt==k-1)
            {
                while(pl!=nex[pos[i]]&&pr<=n)//最左不超过当前这个数的位置，最右不超过n
                {
                    //pl-pre[pl] nex[pr]-pr 之间的差为原始数列中
                    //[pl.pre[pl] ] [next[pr],pr]区间之间小于当前遍历的数的数量
                    ans+=(LL)i*abs(pl-pre[pl])*abs(nex[pr]-pr);
                    printf("i=%d pl=%d pre=%d pr=%d nex=%d\n",i,pl,pre[pl],pr,nex[pr]);
                    pl=nex[pl];
                    pr=nex[pr];
                }
            }
            //从小到大，依次删除元素，比某个数小的数不会影响该数成为第K大
            //删除当前遍历的数
            pl=pre[pos[i]];
            pr=nex[pos[i]];
            pre[pr]=pl;
            nex[pl]=pr;

        }
        printf("%I64d\n",ans);
    }
    return 0;
}