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
,deg
,vis
;
vector<int> e
;
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;
}