HDU6058Kanade's sum（思维+链表模拟）

题目链接

题意：

给出一个n和一个k，求1~n的每个区间的第k大的总和是多少，区间长度小于k的话，贡献为0.

分析：

我们只要求出对于一个数$x$左边最近的$k$个比他大的和右边最近$k$个比他大的，扫一下就可以知道有几个区间的$k$大值是$x$.

我们考虑从小到大枚举$x$，每次维护一个链表，链表里只有$>=x$的数，那么往左往右找只要暴力跳$k$次，删除也是$O(1)$的。

时间复杂度：$O(nk)$

#include<bits/stdc++.h>using namespace std;typedef long long ll;const int N = 500010;int Case, n, K, num[N], pos[N], pre[N], nxt[N];inline void del(int x){    pre[nxt[x]] = pre[x];    nxt[pre[x]] = nxt[x];}inline ll solve(int x){    int a[100], b[100];    int ca = 0, cb = 0;    for(int i = x; i; i = pre[i])    {        a[++ca] = i - pre[i];        if(ca == K) break;    }    for(int i = x; i <= n; i = nxt[i])    {        b[++cb] = nxt[i] - i;        if(cb == K) break;    }    ll t = 0;    for(int i = 1; i <= ca; i++)        if(K - i + 1 <= cb)            t += 1LL * a[i] * b[K-i+1];    return t;}int main(){    scanf("%d", &Case);    while(Case--)    {        scanf("%d%d", &n, &K);        for(int i = 1; i <= n; i++) scanf("%d", &num[i]), pos[num[i]] = i;        for(int i = 0; i <= n+1; i++) pre[i] = i-1, nxt[i] = i+1;        pre[0] = 0;        nxt[n+1] = n+1;        ll ans = 0;        for(int i = 1; i <= n; i++)        {            int x = pos[i];            ans += solve(x) * i;            del(x);        }        printf("%I64d\n", ans);    }    return 0;}