Atcoer Grand Contest 001 F 拓补排序 STL

题意：给定一个1⋯n的排列{pi}，给定一个K，对于pi=pj+1且|i−j|≥K可以交换pi和pj的值，求经过变换可以得到的最小的字典序。

神题= =完全没什么想法，没想到暴力90分，数据水的不行了。
首先做出原序列的对应序列，设为q[i]。
对于q[i]，所有满足$|i-j|<=K且pi<=pj$的，连边求出最小字典序就是答案。
这样子连边复杂度是n^2。
怎么优化呢，我们并不需要任意满足的边都连，只需要连区间内小于该数的数中最大的数，这个很明显了，其余的边对于这个数都没有贡献。
至于处理的话，嫌线段树麻烦可以用set.

#include <algorithm>
#include <cstdio>
#include <cmath>
#include <set>
using namespace std;
const int N=500010;
struct edge{int s,t,n;}e[2*N];
int n,k,tot,p[N],q[N],h[N],d[N];
set <int> S;
set <int> ::iterator It;
void work()
{
    scanf("%d %d",&n,&k);
    for (int i=1; i<=n; i++)  scanf("%d",&p[i]),q[p[i]]=i;
    for (int i=1; i<=n; i++)
        {
            if (i>k)  S.erase(p[i-k]);
            It=S.insert(p[i]).first;
            if ((++It)!=S.end())
                e[++tot]=(edge){i,q[*It],h[i]},h[i]=tot,d[q[*It]]++;
        }
    S.clear();
    for (int i=n; i; i--)
        {
            if (i+k<=n)  S.erase(p[i+k]);
            It=S.insert(p[i]).first;
            if ((++It)!=S.end())
                e[++tot]=(edge){i,q[*It],h[i]},h[i]=tot,d[q[*It]]++;
        }
    S.clear();
    for (int i=1; i<=n; i++)  if (!d[i])  S.insert(i);
    for (int i=1; i<=n; i++)
        {
            It=S.begin(),S.erase(It),p[*It]=i;
            for (int j=h[*It]; j; j=e[j].n)
                if (!(--d[e[j].t]))
                    S.insert(e[j].t);
        }
    for (int i=1; i<=n; i++)  printf("%d\n",p[i]);
}
int main()
{
    freopen("permutation.in","r",stdin);
    freopen("permutation.out","w",stdout);
    work();
    return 0;
}