【模板】前向星 SPFA求最短（长）路

之前一个改自别人的模板竟然在一道题上TLE了，而代码也实在丑陋，网上找得到的模板也大多跑得慢（vector存图）或代码丑陋、残疾（无初始化函数的模板能叫模板吗？），索性自己重新写了一个。

题是POJ1716（差分约束模型），需要求最长路，其实跟最短路也只有很小的差别。

#include<iostream>
#include<cstdio>
#include<vector>
#include<cstring>
#include<queue>
using namespace std;
// 最长路，改为最短路只需修改两个注释了的地方
const int MAXN = 10011, MAXM = 30033;
int N,dis[MAXN],cnt[MAXN],head[MAXN],tot;
bool inq[MAXN];
struct Edge{
    int v, w, nxt;
}e[MAXM];
void init(int n){
    tot = 0; N = n;
    memset(head,-1,sizeof head);
    memset(inq,0,sizeof inq);
    memset(cnt,0,sizeof cnt);
    fill(dis,dis+N+1,-1e9);     /// 1
}
void addEdge(int u,int v,int w){
    e[tot] = {v,w,head[u]};
    head[u] = tot++;
}
bool SPFA(int s){
    queue<int> que;
    inq[s] = 1; dis[s] = 0;
    que.push(s);
    while(!que.empty()){
        int x = que.front(); que.pop();
        inq[x] = 0;
        for(int i = head[x];~i;i = e[i].nxt){
            int v = e[i].v, w = e[i].w;
            if(dis[v] < dis[x] + w){    /// 2
                dis[v] = dis[x] + w;
                if(!inq[v]){
                    inq[v] = 1; que.push(v);
                    if(++cnt[v] > N) return 0;
                }
            }
        }
    }
    return 1;
}

int n;

int main(){
    cin >> n;
    init(10007);
    for(int i = 0;i < n;++i){
        int a,b;
        scanf("%d%d",&a,&b);
        addEdge(a,b+1,2);
    }
    for(int i = 0;i <= 10000;++i){
        addEdge(i,i+1,0);
        addEdge(i+1,i,-1);
    }
    SPFA(0);
    printf("%d
",dis[10001]);

    return 0;
}