hdu 6058 Kanade's sum

/*     计算每个数的贡献，找到左边k-1个大于 pi的，右边 大于pi 的， 所有有贡献的区间都在这个范围内。用链表维护位置，结束后删除pi, 从小到大 枚举，可以保证链表中的数 >=pi; */ #include<iostream>#include<algorithm>#include<cstdio>using namespace std;typedef long long ll;const int mod =1e9+7;const int Size = 510000;int pre[Size],nxt[Size],pos[Size];inline void read(int &x){    x=0;char p=getchar();    while(!(p<='9'&&p>='0')) p=getchar();    while(p<='9'&&p>='0')x*=10,x+=p-48,p=getchar();}int main(){int t,n,k;read(t);while(t--){int p=0,x;read(n);read(k);for(int i=1;i<=n;i++) {    read(x); pos[x]=i; nxt[p]=x; pre[x]=p; p=x; } nxt[x]=n+1;pos[n+1]=n+1; ll ans=0; for(int i=1;i<=n;i++)   {      p=i;      int cnt=1,p1=i;   ll sum=0;      while(cnt<k&&pre[p]) p=pre[p],cnt++;      while(cnt<k&&nxt[p1]!=n+1) p1=nxt[p1],cnt++;      if(cnt<k) continue;      for(;p!=nxt[i];p=nxt[p])        {        int l1=(pos[p]-pos[pre[p]]);        int r1=(pos[nxt[p1]]-pos[p1]);        sum +=l1*r1;        p1=nxt[p1];        if(p1==n+1) break;         }pre[nxt[i]]=pre[i];nxt[pre[i]]=nxt[i]; ans +=(ll)sum*i;   }   printf("%lld\n",ans);}return 0;}