洛谷p2294（前缀和+差分约束）

题目

代码如下：

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e6+5;
const int inf=0x3f3f3f3f;
typedef long long ll;
int n,m;
struct Edge
{
    int from,to,dist;
};
vector<Edge>edges;
vector<int>G[maxn];
void init()
{
    for(int i=0;i<=n;i++)
        G[i].clear();
    edges.clear();
}
void add(int from,int to,int dist)
{
    edges.push_back({from,to,dist});
    int m=edges.size();
    G[from].push_back(m-1);
}
int dis[maxn],cnt[maxn];
bool vis[maxn];
bool spfa()
{
    memset(dis,inf,sizeof(dis));
    memset(cnt,0,sizeof(cnt));
    memset(vis,false,sizeof(vis));
    dis[n+1]=0;
    vis[n+1]=true;
    queue<int>q;
    q.push(n+1);
    while(!q.empty())
    {
        int u=q.front();
        q.pop();
        vis[u]=false;
        for(int i=0;i<G[u].size();i++)
        {
            Edge e=edges[G[u][i]];
            int v=e.to;
            if(dis[v]>dis[u]+e.dist)
            {
                dis[v]=dis[u]+e.dist;
                if(!vis[v])
                {
                    vis[v]=true;
                    cnt[v]++;
                    if(cnt[v]>=n+1)
                        return false;
                    q.push(v);
                }
            }
        }

    }
    return true;
}


int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int w;
    cin>>w;
    while(w--)
    {
        cin>>n>>m;
        init();
        for(int i=0;i<=n;i++)
        {
            add(n+1,i,0);
        }
        for(int i=0;i<m;i++)
        {
            int s,t,v;
            cin>>s>>t>>v;
            add(s-1,t,v);
            add(t,s-1,-v);
        }
        if(!spfa())
            cout<<"false"<<endl;
        else
            cout<<"true"<<endl;
    }


}