HDU6058 Kanade's sum【模拟】

题意：找所有区间的第k大值，没有第k大就是0，求和

思路：我们需要一个链表，记录每个位置左右比他大的数的第一个位置。n从大到小枚举，我们用set存放比它大的数的坐标，查找距离它最近的位置，更新链表。我们在这个位置左右找比它大的k-1个值，那这个位置就是第k大的。找到这些区间即可。

#include<stdio.h>#include<iostream>#include<string.h>#include<string>#include<stdlib.h>#include<math.h>#include<vector>#include<list>#include<map>#include<stack>#include<queue>#include<set>#include<algorithm>#include<numeric>#include<functional>using namespace std;typedef long long ll;const int maxn = 5e5+5;set<int> s;set<int>::iterator iter;int a[maxn],vis[maxn],le[maxn],ri[maxn];int main(void){int n,k,i,j,T;scanf("%d",&T);while(T--){scanf("%d%d",&n,&k);s.clear();memset(le,-1,sizeof le);memset(ri,-1,sizeof ri);for(i = 1; i <= n; i++){scanf("%d",&a[i]);le[i] = 0;ri[i] = n + 1;vis[a[i]] = i;}s.insert(0);s.insert(n+1);le[n+1] = 0;ri[0] = n + 1;long long ans = 0;for(i = n; i >= 1; i--){int l,r;s.insert(vis[i]);iter = s.find(vis[i]);iter++;r = *iter;l = le[r];ri[l] = vis[i];ri[vis[i]] = r;le[r] = vis[i];le[vis[i]] = l;int nowl,nowr;nowl = vis[i];int cnt = 0;while(cnt < k && nowl != 0){nowl = le[nowl];cnt++;}nowr = vis[i];while(cnt < k && nowr != n+1){nowr = ri[nowr];cnt++;}if(cnt < k)continue;int nl = ri[nowl],nr = ri[nowr];while(nowl < vis[i] && nowr < n+1){ans += 1ll * (nl-nowl) * (nr - nowr) * i;nowl = ri[nowl];nowr = ri[nowr];nl = ri[nowl],nr = ri[nowr];}}printf("%lld\n",ans);}return 0;}