HDU 5195 DZY Loves Topological Sorting 拓扑序(贪心)

HDU 5195
题意:n个点 m条有向边的DAG, 现在生成[1..n]的排列,若有边u->v,则说明在排列中u出现在v之前.
n,m,k<=1e5,现在问删除不超过k条边的情况下,能生成最大字典序的排列为?


靠前的尽量大,把当前位置能选的点都加入到队列中,即入度<=k的都压倒优先队列中.
每次弹出一个最大的 并更新k,和其他定点的度. 若该点度数>k或者已经被访问,则继续下一个元素.

删除u时遍历u的所有边 每条边都只会被遍历一次 复杂度应该在O(m+n)左右.

还有一种解法用线段树找当前定点度数<=k的最大值 然后更新...

#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> ii;
const int N=2e5+5;
int n,m,k,num,a[N],deg[N],vis[N];
vector<int> e[N];
struct node{
    int val,ned;
    node()
    {

    }
    node(int a,int b)
    {
        val=a;
        ned=b;
    }
    bool operator <(const node &t)const
    {
        if(val==t.val)
            return  ned>t.ned;
        return val<t.val;
    }
};
priority_queue<node> q;
void Topo()
{
    while(!q.empty())
        q.pop();
    for(int i=1;i<=n;i++)
    {
       if(deg[i]<=k)
            q.push(node(i,deg[i]));
    }
    while(!q.empty()&&num<n)
    {
        node t=q.top();
        q.pop();
        int u=t.val,x=t.ned;
        if(x>k||vis[u])
            continue;
        k-=x,a[++num]=u,vis[u]=1;
        for(int i=0;i<e[u].size();i++)
        {
            int v=e[u][i];
            deg[v]--;
            if(deg[v]<=k)
                q.push(node(v,deg[v]));
        }
    }

}
int main()
{
    while(cin>>n>>m>>k)
    {
        memset(deg,0,sizeof(deg));
        memset(vis,0,sizeof(vis));
        for(int i=1;i<=n;i++)
            e[i].clear();
        num=0;
        int u,v;
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&u,&v);
            e[u].push_back(v);
            deg[v]++;
        }
        Topo();
        for(int i=1;i<=n;i++)
            printf("%d%c",a[i],i==n?'\n':' ');
    }
    return 0;
}