A*求第k短路求调WA#3#6
查看原帖
A*求第k短路求调WA#3#6
399650
STUDYING0601楼主2022/9/29 10:58
#include<bits/stdc++.h>
using namespace std;
const int Mx=1e5+5;
int n,m;
struct node{
    int u,v,next;
    double w;
}e[Mx<<1];
int head[Mx],tot=0;double dis_t[Mx],g[Mx];int cnt[Mx],f[Mx];
void add(int u,int v,double w){
    ++tot;
    e[tot].u=u;
    e[tot].v=v;
    e[tot].w=w;
    e[tot].next=head[u];
    head[u]=tot;
}
struct point{
    int x,y;
}a[Mx];
double dis(point a,point b){
    return sqrt((a.x-b.x)*(a.x-b.x)+(a.y-b.y)*(a.y-b.y));
}
struct dj{
    int u;double dis;
    friend bool operator<(dj i,dj j){
        return i.dis>j.dis;
    }
};
struct A{
    int u;double g;
    int vis[205];
    A(){
        memset(vis,0,sizeof(vis));
    }
    friend bool operator<(A i,A j){
        return i.g+dis_t[i.u]>j.g+dis_t[j.u];
    }
};

int main(){
    cin>>n>>m;
    for(int i=1;i<=n;i++)cin>>a[i].x>>a[i].y;
    for(int i=1,u,v;i<=m;i++){
        cin>>u>>v;
        double w=dis(a[u],a[v]);
        add(u,v,w);
        add(v,u,w);
    }
    
    priority_queue<dj> q1;
    for(int i=1;i<=n;i++)dis_t[i]=2147483647;
    q1.push((dj){n,0});
    dis_t[n]=0;
    while(!q1.empty()){
        dj tmp=q1.top();q1.pop();
        int u=tmp.u;
        if(f[u])continue;
        f[u]=1;
        for(int i=head[u];i;i=e[i].next){
            int v=e[i].v;double w=e[i].w;
            if(dis_t[u]+w<=dis_t[v]){
                dis_t[v]=dis_t[u]+w;
                q1.push((dj){v,dis_t[v]});
            }
        }
    }
    priority_queue<A> q2;
    A st;st.g=0,st.u=1;
    q2.push(st);
    while(!q2.empty()){
        A tmp=q2.top();q2.pop();
        int u=tmp.u;
        cnt[u]++;
        if(u==n&&cnt[u]==2){
            cout<<fixed<<setprecision(2)<<tmp.g<<endl;
            return 0;
        }
        for(int i=head[u];i;i=e[i].next){
            int v=e[i].v;double w=e[i].w;
            if(tmp.vis[v])continue;
            A nxt;nxt.g=tmp.g+w;nxt.u=v;nxt.vis[v]=1;
            q2.push(nxt);
        }
    }
    cout<<-1<<endl;
    return 0;
}
2022/9/29 10:58
加载中...