dfs tle on #9求助
查看原帖
dfs tle on #9求助
747882
_adil_楼主2023/10/2 11:16

本地可以得到答案,没有死循环...

#include<bits/stdc++.h>
#define N 12009
#define INF 0x3f3f3f3f
#define mod 998244353
#define int long long
using namespace std;
typedef long double ldb;
typedef pair<int,int> pii;
int t,n,m,k,cnt;
int head[N],to[N],val[N],nxt[N],vis[2009],dis[2009];
void add(int u,int v,int w){
  to[++cnt]=v,nxt[cnt]=head[u],head[u]=cnt,val[cnt]=w;
}
int dfs(int u){
  vis[u]=1;
  for(int i=head[u];i;i=nxt[i]){
    int v=to[i],w=val[i];
    if(dis[v]>dis[u]+w){
      if(vis[v]){return 1;}
      dis[v]=dis[u]+w;
      if(dfs(v))return 1;
    }
  }
  vis[u]=0;
  return 0;
}
signed main(){
  ios::sync_with_stdio(false);
  cin.tie(0);
  cout.tie(0);
  cin>>t;
  for(int j=0;j<t;j++){
    cin>>n>>m;
    memset(vis,0,sizeof(vis)),memset(dis,INF,sizeof(dis)),dis[1]=0,cnt=0;
    memset(head,0,sizeof(head)),memset(to,0,sizeof(to)),memset(val,0,sizeof(val)),memset(nxt,0,sizeof(nxt));
    for(int i=1;i<=m;i++){
      int u,v,w;
      cin>>u>>v>>w;
      if(w>=0)add(u,v,w),add(v,u,w);
      else add(u,v,w);
    }
    if(dfs(1))cout<<"YES\n";
    else cout<<"NO\n";
  }
  return 0;
}
2023/10/2 11:16
加载中...