90pts 求助帖
查看原帖
90pts 求助帖
1015796
Lingyu_fly楼主2024/10/16 12:42
#include<bits/stdc++.h>
using namespace std;
typedef pair<int,int> pii;
const int N=1e4+10;
int n,m,ans;
int dis[N],vis[N],danger[N],book[N];
struct edge{
    int v,w;
};
vector<edge> e[N];
int dijkstra(int s,int x){
    priority_queue<pii> q;
    memset(dis,0x3f,sizeof dis);
    memset(vis,0,sizeof vis);
    dis[s]=0;
    q.push({0,s});
    while(q.size()){
        auto t=q.top();q.pop();
        int u=t.second;
        if(vis[u]) continue;
        vis[u]=1;
        for(auto i:e[u]){
            int v=i.v,w=i.w;
            if(dis[v]>dis[u]+w){
                dis[v]=dis[u]+w;
                q.push({-dis[v],v});
            }
        }
    }
    return dis[x];
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        scanf("%d",&danger[i]);
        book[danger[i]]=1;
    }
    for(int i=1;i<=n;i++){
        for(int j=1;j<=n;j++){
            int w;
            scanf("%d",&w);
            e[i].push_back({j,w});
        }
    }
    for(int i=1;i<m;i++){
        ans+=dijkstra(danger[i],danger[i+1]);
    }
    danger[m+1]=danger[m];
    for(int i=1;i<=n;i++){
        if(danger[i]==0){
            ans+=dijkstra(danger[m+1],i);
            danger[m+1]=i;
        }
    }
    printf("%d",ans);
    return 0;
}

没走的点也判断过了

2024/10/16 12:42
加载中...