HDU 6058 Kanade's sum（链表）

 http://acm.hdu.edu.cn/showproblem.php?pid=6058

题意：
找出所有区间第K大的值之和。

思路：

又有点贡献值的味道，就是考虑当前这个数贡献了几个区间。

然后往左和往右分别找大于当前数的k-1个数，这样就可以确定区间的个数，这要求我们从小到大找 并且找完之后需要删除这个数，用链表来维护。

删除元素的目的是为了加速，保证了当前查找的元素是最小值，所以只需要跳跃寻找k次就可以。具体看代码。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#include<sstream>
#include<vector>
#include<stack>
#include<queue>
#include<cmath>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
typedef pair<int,int> pll;
const int INF = 0x3f3f3f3f;
const int maxn = 5*1e5 + 5;

int n, k;

int a[maxn];
int pre[maxn];
int nxt[maxn];
int pos[maxn];

int before[maxn];
int after[maxn];

ll solve(int p)
{
    int c1=0,c2=0;
    for(int i=p; i&&c1<=k; i=pre[i])
    {
        before[c1++]=i-pre[i];  //左边i个比当前数大的方法数
    }
    for(int i=p; i<=n&&c2<=k; i=nxt[i])
    {
        after[c2++]=nxt[i]-i;
    }

    ll ans=0;
    for(int i=0;i<c1;i++)  //左边i个，右边k-i+1个
    {
        if(k-i-1<c2)
        {
            ans+=before[i]*after[k-i-1];
        }
    }

    pre[nxt[p]]=pre[p];
    nxt[pre[p]]=nxt[p];

    return ans;
}

int main()
{
    //freopen("in.txt","r",stdin);
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            pos[a[i]]=i;
        }
        for(int i=1;i<=n;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++)
        {
            ans+=solve(pos[i])*i;
        }
        printf("%lld
",ans);
    }
    return 0;
}