hdu6058 Kanade's sum

题解:找出x左右两边大于x的k个数，左边记为cntl右边记为cntr，当cntl+cntr==k时，开始统计有多少个区间以x为第k大值。

重点是从小到大找x,x的情况全部考虑完后，把x从链表里删掉。

复杂度为O(n*k)

#include<iostream>
#include<stdio.h>
using namespace std;
typedef long long ll;
const int maxn=1e6;
int t,n,k,id[maxn];
struct node{
    int nxt,pre,id;
}buf[maxn];
ll ans;
int main(){
    scanf("%d",&t);
    while(t--){
        ans=0;
        scanf("%d%d",&n,&k);
        for(int x,i=1;i<=n;i++){
            scanf("%d",&x);
            id[x]=i;
            buf[i]=node{i+1,i-1,i};
        }
        buf[n+1].id=n+1;
        for(int i=1;i<=n;i++){
            int l=id[i],r=id[i];
            int cntl=1,cntr=0;
            while(cntl<k){
                if(buf[l].pre==0) break;
                l=buf[l].pre;
                cntl++;
            }
            while(cntl){
                while(cntl+cntr<k){
                    if(buf[r].nxt==n+1) break;
                    r=buf[r].nxt;
                    cntr++;
                }
                while(cntl+cntr>k){
                    cntr--;
                    r=buf[r].pre;
                }
                if(cntl+cntr==k){
                    int L=buf[l].id-buf[buf[l].pre].id;
                    int R=buf[buf[r].nxt].id-buf[r].id;
                    ans+=(ll)L*R*i;
                }
                l=buf[l].nxt,cntl--;
            }
            buf[buf[id[i]].pre].nxt=buf[id[i]].nxt;
            buf[buf[id[i]].nxt].pre=buf[id[i]].pre;
        }
        printf("%lld\n",ans);
    }
}