HDU 5195 DZY Loves Topological Sorting（线段树）

看题意以为是个拓扑排序。事实上，就是个线段树。因为最多可以删k条边， 所以就是在线段树里找入度小于等于k的最大值，那么保存个区间最小就ok了。如果右子树的区间最小小于等于k那么就往右边走，因为是要找字典序最大的。当k是0，也是这样找，就跟topological sort是一样的。然后删除一个点就在线段树那个点置最大值，再删除他的边。因为总共也就m条边不超过10^5。

总复杂度o(n+m)logn。

AC代码：

#pragma comment(linker, "/STACK:102400000,102400000")
#include<cstdio>
#include<ctype.h>
#include<algorithm>
#include<iostream>
#include<cstring>
#include<vector>
#include<cstdlib>
#include<stack>
#include<queue>
#include<set>
#include<map>
#include<cmath>
#include<ctime>
#include<string.h>
#include<string>
#include<sstream>
#include<bitset>
using namespace std;
#define ll __int64
#define ull unsigned long long
#define eps 1e-8
#define NMAX 1000000000
#define MOD 1000000
#define lson l,mid,rt<<1
#define rson mid+1,r,rt<<1|1
#define PI acos(-1)
template<class T>
inline void scan_d(T &ret)
{
    char c;
    int flag = 0;
    ret=0;
    while(((c=getchar())<'0'||c>'9')&&c!='-');
    if(c == '-')
    {
        flag = 1;
        c = getchar();
    }
    while(c>='0'&&c<='9') ret=ret*10+(c-'0'),c=getchar();
    if(flag) ret = -ret;
}
template<class T> inline T Max(T a, T b){ return a > b ? a : b; }
template<class T> inline T Min(T a, T b){ return a < b ? a : b; }

const int maxn = 100000+10;
int T[maxn<<2],a[maxn],head[maxn],nct;
int du[maxn];
struct Edge
{
    int v,next;
}e[maxn];

void add(int u, int v)
{
    e[nct].v = v; e[nct].next = head[u];
    head[u] = nct++;
}

inline void pushup(int rt)
{
    T[rt] = min(T[rt<<1],T[rt<<1|1]);
}

void build(int l, int r, int rt)
{
    if(l == r)
    {
        T[rt] = du[l];
        return;
    }
    int mid = (l+r)>>1;
    build(lson);
    build(rson);
    pushup(rt);
}

int query(int k, int l, int r, int rt)
{
    if(l == r)
    {
        T[rt] = 100000+10;
        return l;
    }
    int mid = (l+r)>>1,m;
    if(T[rt<<1|1] <= k) m = query(k,rson);
    else m = query(k,lson);
    pushup(rt);
    return m;
}

void update(int L, int k, int l, int r, int rt)
{
    if(l == r)
    {
        T[rt] -= k;
        return;
    }
    int mid = (l+r)>>1;
    if(L <= mid) update(L,k,lson);
    else update(L,k,rson);
    pushup(rt);
}

int main()
{
#ifdef GLQ
    freopen("input.txt","r",stdin);
//    freopen("o.txt","w",stdout);
#endif
    int n,m,k;
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        nct = 0;
        memset(du,0,sizeof(du));
        memset(head,-1,sizeof(head));
        for(int i = 1; i <= m; i++)
        {
            int u,v;
            scanf("%d%d",&u,&v);
            add(u,v);
            du[v]++;
        }
        build(1,n,1);
        for(int i = 1; i <= n; i++)
        {
            int w = query(k,1,n,1);
            printf("%d%c",w,(i==n)?'\n':' ');
            k -= du[w];
            du[w] = 0;
            for(int j = head[w]; j != -1; j = e[j].next)
            {
                int v = e[j].v;
                if(du[v])
                {
                    du[v]--;
                    update(v,1,1,n,1);
                }
            }
        }
    }
    return 0;
}