MnZn求助spfa判负环
查看原帖
MnZn求助spfa判负环
339846
RuntimeErr楼主2021/6/5 14:44

RT,TLE on #8

开了 O2 就过了

#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;

const int N=510,M=5510,inf=0x7ffffff;

int n,m,k;
int h[N],e[M],ne[M],w[M],cnt[N],dis[N],idx;
bool vis[N];
queue<int>q;

inline void add(int a,int b,int c){
    e[++idx]=b;ne[idx]=h[a];w[idx]=c;h[a]=idx;
}

bool spfa(int s) {
    for(int i=1;i<=n;++i)dis[i]=inf,vis[i]=cnt[i]=0;
    dis[s]=0;vis[s]=true;q.push(s);
    while(!q.empty()) {
        int u=q.front();q.pop();
        vis[u]=false;
        for(int i=h[u];i;i=ne[i]){
            int v=e[i];
            if(dis[v]>dis[u]+w[i]){
                dis[v]=dis[u]+w[i];
                if(!vis[v]){
                    ++cnt[v];
                    vis[v]=true;
                    q.push(v);
                }
                if(cnt[v]==n)return true;
            }
        }
    }
    return false;
}
int main(){
    int t;scanf("%d",&t);
    while(t--){
        scanf("%d%d%d",&n,&m,&k);
        idx=0;memset(h,0,sizeof h);
        for(int i=1;i<=m;++i){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);add(b,a,c);
        }
        for(int i=1;i<=k;++i){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,-c);
        }
        if(spfa(1))puts("YES");
        else puts("NO");
    }
    return 0;
}
2021/6/5 14:44
加载中...