题解 P3074 【[USACO13FEB]牛奶调度Milk Scheduling】

Datura_ · 2018-08-25 17:37:31 · 题解

拓扑排序模板题。

这里我用了广搜的删边法，记录每个点的入度，从入度为0的点开始搜索，每删一条边入度减1。

AC代码：

#include<iostream>
#include<vector>
#include<queue>
using namespace std;

const int MAXN=10001;
int n,m,ans,t[MAXN],tot[MAXN],d[MAXN];
vector<int> g[MAXN];
queue<int> q;

int main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
        cin>>t[i];
    for(int i=1;i<=m;i++)
    {
        int x,y;
        cin>>x>>y;
        g[x].push_back(y);
        d[y]++;//记录入度
    }
    for(int i=1;i<=n;i++)
    {
        if(d[i]==0)//入度为0的点入队
        {
            tot[i]=t[i];
            q.push(i);
        }
    }
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        for(int i=0;i<g[u].size();i++)
        {
            int v=g[u][i];
            d[v]--;//删边
            tot[v]=max(tot[v],tot[u]+t[v]);
            if(d[v]==0)q.push(v);//入度为0的点入队
        }
    }
    for(int i=1;i<=n;i++)
        ans=max(tot[i],ans);
    cout<<ans;
    return 0;
}