lfyzoj1096 最短路径问题 dijkstra堆优化

Posted 2020-10-16 poorpool

题目描述

平面上有n个点（n<=100），每个点的坐标均在-10000~10000之间。其中的一些点之间有连线。

若有连线，则表示可从一个点到达另一个点，即两点间有通路，通路的距离为两点间的直线距离。现在的任务是找出从一点到另一点之间的最短路径。

输入

第一行为整数n。

第2行到第n+1行（共n行） ，每行两个整数x和y，描述了一个点的坐标。

第n+2行为一个整数m，表示图中连线的个数。

此后的m 行，每行描述一条连线，由两个整数i和j组成，表示第i个点和第j个点之间有连线。

最后一行：两个整数s和t，分别表示源点和目标点。

输出

仅一行，一个实数（保留两位小数），表示从s到t的最短路径长度。

样例输入

5
0 0
2 0
2 2
0 2
3 1
5
1 2
1 3
1 4
2 5
3 5
1 5

样例输出

3.41

这题根本不用堆优化，练习一下而已

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdio>
#include <cmath>
using namespace std;
struct Edge{
    int too, nxt;
    double val;
}edge[10005];
int n, m, uu, vv, s, t, k, d[105], cnt, hea[105];
double xx[105], yy[105], dis[105];
bool vis[105];
bool cmp(int x, int y){
    return dis[x]>dis[y];
}
void add_edge(int fro, int too, double val){
    edge[++cnt].nxt = hea[fro];
    edge[cnt].too = too;
    edge[cnt].val = val;
    hea[fro] = cnt;
}
int main(){
    cin>>n;
    for(int i=1; i<=n; i++){
        scanf("%d %d", &uu, &vv);
        xx[i] = uu;
        yy[i] = vv;
    }
    cin>>m;
    for(int i=1; i<=m; i++){
        scanf("%d %d", &uu, &vv);
        double sid=sqrt((xx[uu]-xx[vv])*(xx[uu]-xx[vv])+(yy[uu]-yy[vv])*(yy[uu]-yy[vv]));
        add_edge(uu, vv, sid);
        add_edge(vv, uu, sid);
    }
    cin>>s>>t;
    memset(dis, 127, sizeof(dis));
    dis[s] = 0.0;
    d[++k] = s;
    while(k){
        int j=d[1];
        pop_heap(d+1, d+1+k, cmp);
        k--;
        if(vis[j])  continue;
        vis[j] = true;
        if(j==t)    break;
        for(int i=hea[j]; i; i=edge[i].nxt){
            int t=edge[i].too;
            if(!vis[t] && dis[t]>dis[j]+edge[i].val){
                dis[t] = dis[j] + edge[i].val;
                d[++k] = t;
                push_heap(d+1, d+1+k, cmp);
            }
        }
    }
    printf("%.2lf\n", dis[t]);
    return 0;
}