HDU 6058 Kanade's sum 链表+枚举

    题意是说给出n个数，这n个数为[1,n]打乱顺序放入，求出这些数构成的所有区间的第k大数的和。

    要直接去做肯定不行，比赛时候主席树、找规律各种方法都用上还是TLE没能过，比完赛之后看标程才知道这题可以用链表来存储做这道题目。

    首先读入数据，然后记录每一个数的位置，用数组来构建一个双向链表，然后从1开始找，因为1是最小的，所以往左往右找数，只要构成一个长度为k的区间，1就是第k大的数。之后将1从删掉，再去找2，依然往两边去扩散区间，为了防止1删去的影响，应用他们的位置相减来得到结果，一直找到n-k+1，就能得到所有数的结果。

    程序中用pos数组来记录位置，pre和np是用来模拟双向链表的过程进行指向，之后的t和s数组来记录下位置，之后往左寻找x个数，则就是要往右找k-x-1个数。

    下面AC代码：（基本是改于标程= =）

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
int a[1000005];
int pos[1000005];
int s[1000005];
int t[1000005];
int pre[1000005];
int np[1000005];
int n;

int eras(int x)
{
    int pp=pre[x];
    int nn=np[x];
    if(pre[x])
        np[pre[x]]=nn;
    if(np[x]<=n)
        pre[np[x]]=pp;
    pre[x]=np[x]=0;
    return 0;
}

int main()
{
    int T;
    int i,j,d;
    long long ans;
    int k;
    int p;
    int ss,tt;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&k);
        memset(pre,0,sizeof(pre));
        memset(np,0,sizeof(np));
        for(i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            pos[a[i]]=i;
        }
        for(i=1;i<=n;i++)
        {
            pre[i]=i-1;
            np[i]=i+1;
        }
        ans=0;
        for(j=1;j<=n-k+1;j++)
        {
            p=pos[j];
            ss=tt=0;
            for(d=p;d!=0&&ss<=k+1;d=pre[d])
            {
                ss++;
                s[ss]=d;
            }
            for(d=p;d!=n+1&&tt<=k+1;d=np[d])
            {
                tt++;
                t[tt]=d;
            }
            ss++;
            tt++;
            s[ss]=0;
            t[tt]=n+1;
            for(i=1;i<=ss-1;i++)
            {
                if(k+1-i<=tt-1&&k+1-i>=1)
                {
                    ans+=(long long)(t[k+1-i+1]-t[k+1-i])*(long long)(s[i]-s[i+1])*(long long)j;
                    //cout<<ans<<"+=(long long)("<<t[k+1-i+1]<<"-"<<t[k+1-i]<<")*(long long)("<<s[i]<<"-"<<s[i+1]<<")*"<<j<<endl;
                }
            }
            eras(p);
        }
        cout<<ans<<endl;
    }
    return 0;
}