Description
给出一个n个点m条边的有向无环图,要求删去k条边后使得该图的最小字典序拓扑序的字典序尽可能小
Input
第一行一整数T表示用例组数,每组用例首先输入三个整数n,m,k分别表示点数,边数以及要删去的边数,之后m行每行两个整数u和v表示u到v有一条有向边(1<=n<=1e5,0<=k<=m<=2e5)
Output
对于每组用例,设删去k条边后最小拓扑序为p1,…,pn,输出sum{i*pi} mod 1e9+7
Sample Input
3
4 2 0
1 2
1 3
4 5 1
2 1
3 1
4 1
2 3
2 4
4 4 2
1 2
2 3
3 4
1 4
Sample Output
30
27
30
Solution
贪心的往前面放编号尽可能小的点,开一个优先队列,从前往后一个个往拓扑序列中插点,每次从优先队列中取点(即当前剩余点中编号最小的点),如果其入度不大于k(此处入度表示没进拓扑序列中的点对该点入度的贡献,k表示还能删去的边数),那么显然删去以其为终点的边,即把该点放在拓扑序列当前位置是最优的
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
#include<vector>
#include<queue>
using namespace std;
typedef long long ll;
#define mod 1000000007ll
#define maxn 111111
priority_queue< int,vector<int>,greater<int> >que;
vector<int>g[maxn];
int T,n,m,k,in[maxn],vis[maxn];
int main()
{
scanf("%d",&T);
while(T--)
{
scanf("%d%d%d",&n,&m,&k);
while(!que.empty())que.pop();
for(int i=1;i<=n;i++)in[i]=0,vis[i]=0,que.push(i),g[i].clear();
while(m--)
{
int u,v;
scanf("%d%d",&u,&v);
g[u].push_back(v),in[v]++;
}
int res=1;
ll ans=0;
while(!que.empty())
{
int u=que.top();
que.pop();
if(vis[u])continue;
if(in[u]<=k)
{
vis[u]=1;
k-=in[u];
ans=(ans+1ll*res*u%mod)%mod;
res++;
for(int i=0;i<g[u].size();i++)
{
int v=g[u][i];
if(!vis[v])que.push(v),in[v]--;
}
}
}
printf("%I64d\n",ans);
}
return 0;
}