为什么dij加上vis更慢了?
查看原帖
为什么dij加上vis更慢了?
462558
ziyistudy楼主2024/10/25 18:34

加 vis 数组:1.24s,最慢146ms 不加 vis 数组:1.01s,最慢 121ms

#include <bits/stdc++.h>
using namespace std;
#define int long long
#define pb push_back
#define eb emplace_back
#define all(x) (x).begin(),(x).end()
#define fi first
#define se second
#define SZ(x) ((int)(x).size())
typedef vector<int> VI;
typedef long long ll;
typedef pair<int,int> PII;
const ll MOD=998244353;
// head
const int N=1e4+5,M=105,INF=1e10;
int n,m,k;
vector<vector<PII>> G(N);
int dis[N][M];
int vis[N][M];
void dij()
{
    memset(dis,0x3f,sizeof(dis));
    priority_queue<PII,vector<PII>,greater<PII>> Q;
    Q.push({0,1});
    dis[1][0]=0;
    while(!Q.empty()){
        int u=Q.top().se,t=Q.top().fi;
        Q.pop();
        // if(vis[u][t%k]) continue;
        // vis[u][t%k]=1;
        for(auto it:G[u]){
            int v=it.fi,T=it.se;
            int tt=t;
            if(tt<T) tt=tt+(T-tt)/k*k+((T-tt)%k>0)*k;
            if(dis[v][(tt+1)%k]>tt+1){
                dis[v][(tt+1)%k]=tt+1;
                Q.push({tt+1,v});
            }
        }
    }
}
signed main()
{
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    cin>>n>>m>>k;
    for(int i=1;i<=m;i++){
        int u,v,w;cin>>u>>v>>w;
        G[u].pb({v,w});
    }

    dij();

    if(dis[n][0]>=INF) cout<<-1<<endl;
    else cout<<dis[n][0]<<endl;
}
2024/10/25 18:34
加载中...