洛谷 P1807 最长路

题目描述

设 G 为有 n 个顶点的带权有向无环图，G 中各顶点的编号为 1 到 n，请设计算法，计算图 G 中 1,n 间的最长路径。

输入格式

输入的第一行有两个整数，分别代表图的点数 n 和边数 m。

第 2 到第 (m+1) 行，每行 3 个整数 u,v,w（u<v），代表存在一条从 u 到 v 边权为 w 的边。

输出格式

输出一行一个整数，代表 1 到 n 的最长路。

若 1 无法到达 n，请输出 −1。

输入输出样例

输入 #1复制

2 1
1 2 1

输出 #1复制

1

说明/提示

【数据规模与约定】

• 对于 20%的数据，n≤100，m≤103。
• 对于 40% 的数据，n≤103，m≤104。
• 对于 100% 的数据，1≤n≤1500，0≤m≤5×104，1≤u,v≤n，−105≤w≤105。

#include<bits/stdc++.h>
using namespace std;
const int N=1510;
typedef pair<int,int> PII;
vector<PII> edges[N];
int n,m;
int in[N];
int f[N];
int main()
{
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int u,v,w;
        cin>>u>>v>>w;
        edges[u].push_back({v,w});
        in[v]++;
    }
    queue<int> q;
    //先处理入度为0，但是不是1号结点的节点
    for(int i=2;i<=n;i++)
    {
        if(in[i]==0)
        q.push(i);
        f[i]=-1e9;
    }
    while(q.size())
    {
        auto z=q.front();
        q.pop();
        for(auto& t:edges[z])
        {
            int v=t.first;
            in[v]--;
            if(in[v]==0)
            q.push(v);
        }
    }
    q.push(1);
     while(q.size())
    {
        auto z=q.front();
        q.pop();
        for(auto& t:edges[z])
        {
            int v=t.first,w=t.second;
            f[v]=max(f[v],w+f[z]);
            in[v]--;
            if(in[v]==0)
            q.push(v);
        }
    }
    if(f[n]==-1e9)
    cout<<-1<<endl;
    else
    cout<<f[n]<<endl;
    return 0;
}