HDU 5638 Toposort(贪心+优先队列+拓扑排序)

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
#include
#include
#include
#include
#include
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;iint v=g[u][i];
                    if(!vis[v])que.push(v),in[v]--;
                }
            }
        }
        printf("%I64d\n",ans);
    }
    return 0;
}

你可能感兴趣的:(HDU,贪心,杂题)