最短路模板整理

Dijkstra

模板一(map数组模拟邻接表)

处理小图速度相对较快。
内存占用较小,对重边优化较差。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>

using namespace std;
const int INF = 0x3f3f3f3f;
const int maxn = 1000;

int map[maxn][maxn];
int pre[maxn],dis[maxn];
bool vis[maxn];
int n,m;

void Dijkstra(int s)
{
    memset(dis,0x3f,sizeof(dis));
    memset(pre,-1,sizeof(pre));
    memset(vis,false,sizeof(vis));
    for(int i=1;i<=n;++i)
    {
        dis[i]=map[s][i];
        pre[i]=s;
    }
    dis[s]=0;
    vis[s]=true;
    for(int i=2;i<=n;++i)
    {
        int mindist=INF;
        int u=s;
        for(int j=1;j<=n;++j)
            if((!vis[j])&&dis[j]<mindist)
            {
                u=j;
                mindist=dis[j];
            }
        vis[u]=true;
        for(int j=1;j<=n;++j)
            if((!vis[j])&&map[u][j]<INF)
            {
                if(map[u][j]+dis[u]<dis[j])
                {
                    dis[j]=map[u][j]+dis[u];
                    pre[j]=u;
                }
            }
    }
}
int main(){
    cin >> m >> n;
    for (int i=1; i <=n;i++)
        for (int j=1;j <=n;j++){
            if (i == j) map[i][j] = 0;
            else map[i][j] = map[j][i] = INF;
        }
    while (m--) {
        int start, end, len;
        cin >> start >> end >> len;
        map[start][end] = len;
        map[end][start] = len;
    }
    cout << map[2][5] << endl;
    Dijkstra(1);
    cout << dis[n] << endl;
    return 0;
}

模板二(链式前向星+优先队列优化)

主要优化在重边。因为使用了STL所以占用内存和速度相对较慢。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<queue>

using namespace std;
const int INF=0x3f3f3f3f;

struct node
{
    int d,u;
    friend bool operator<(node a,node b)
    {
        return a.d>b.d;
    }
    node(int dist,int point):d(dist),u(point){}
};

struct Edge
{
    int to,next;
    int dist;
}edge[maxm];
int head[maxn],tot;
int pre[maxn],dis[maxn];

void init()
{
    memset(head,-1,sizeof(head));
    tot=0;
}

void addedge(int u,int v,int d)
{
    edge[tot].to=v;
    edge[tot].dist=d;
    edge[tot].next=head[u];
    head[u]=tot++;
}

void Dijkstra(int s)
{
    priority_queue<node> q;
    memset(dis,0x3f,sizeof(dis));
    memset(pre,-1,sizeof(pre));
    dis[s]=0;
    while(!q.empty())
        q.pop();
    node a(0,s);
    q.push(a);       //起点入队列
    while(!q.empty())
    {
        node x=q.top();
        q.pop();
        if(dis[x.u]<x.d)   //最短路已找到
            continue;
        for(int i=head[x.u];i!=-1;i=edge[i].next)
        {
            int v=edge[i].to;
            if(dis[v]>dis[x.u]+edge[i].dist)
            {
                dis[v]=dis[x.u]+edge[i].dist;
                pre[v]=x.u;
                q.push(node(dis[v],v));
            }
        }
    }
}

模板三(结构体内置方法)

因为使用了链式前向星所以不担心重边。
其中还使用了快读方法,所以很快。

#include <cstdio>
#include <cstring>
#include <queue>
#include <algorithm>
#include <utility>
using namespace std;

const int N = 1005;
const int M = 1005;
const int INF = 0x3f3f3f3f

struct Graph{
    struct Edge{
        int v, w, next;
    }edge[M];
    int ehead[N];
    void init(){
        memset(ehead, -1, sizeof(ehead))
    }
    inline void addedge(int u, int v, int w){
        edge[ecnt] = {v, w, ehead[u]};
        ehead[u] = ecnt++;
    }
    int dist[N];
    bool vis[N];
    void Dijkstra(int s){
        memset(dist, INF, sizeof(dist));
        memset(vis, 0, sizeof(vis));
        priority_queue<pair<int, int> > q;
        q.push(make_pair(-(dist[s] = 0), s));
        while(q.size()){
            int u = q.top().second; q.pop();
            if (vis[u]) continue;
            vis[u] = true;
            for (int i = ehead[u]; ~i; i = edge[i].next){
                int v = edge[i].v;
                if (vis[v]) continue;
                int ndist = dist[u] + edge[i].w;
                if (ndist < dist[v]) q.push(make_pair(-(dist[v] = ndist), v));
            }
        }
    }
}g1, g2;

int Input(){
    char c;
    for (c = getchar(); c<'0' || c>'9'; c = getchar());
    int a = c - '0';
    for (c = getchar(); c >= '0' && c <= '9'; c = getchar())
        a = a*10 + c - '0';
    return a;
}

Floyd

代码很短,时间复杂度很高(O(n^3))。

void floyd(){
    for(int k=1; k<=n; ++k)
        for(int i=1; i<=n; ++i)
            for(int j=1; j<=n; ++j)
                if (map[i][j] > map[i][k] + map[k][j]) //松弛
                    map[i][j] = map[i][k] + map[k][j];
}

Bellman-Ford

模板一(链式前向星)

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;
struct Edge{
    int u, v;
    double r, c;
}edge[maxn*2];

double mostMoney[maxn];
int n, m, s;
double v;
int tot;

void addedge(int u, int v, double r, double c){
    edge[tot].u = u;
    edge[tot].v = v;
    edge[tot].r = r;
    edge[tot++].c = c;
}

bool relax(int n){
    double temp = (mostMoney[edge[n].u] - edge[n].c)*edge[n].r;
    if (temp > mostMoney[edge[n].v]){
        mostMoney[edge[n].v] = temp;
        return true;
    }
    return false;
}

bool bellman_ford(){
    bool flag;
    for (int i=0; i<n; i++) mostMoney[i] = 0.0;
    mostMoney[s] = v;
    for (int i=0; i<n-1; ++i){
        flag = false;
        for (int j=0; j<tot; ++j)
            if (relax(j)) flag = true;
        if (mostMoney[s] > v) return true;
        if (!flag) return false;
    }
    for (int i=0; i<tot; ++i){
        if (relax(i)) return true;
    }
    return false;
}

模板二

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
using namespace std;

const int INF=0x3f3f3f3f;
struct Edge{
    int u,v; //起点、终点
    int dist; //长度
}edge[maxn];

int dis[maxn]; //最短距离数组
int n, m; //结点数、边数

bool Bellman_ford(int s){
    memset(dis, INF, sizeof(dis));
    dis[s]=0;
    for(int k=1; k<n; ++k){ //迭代n-1次
        for(int i=0; i<m; ++i){  //检查每条边
            int x = edge[i].u, y = edge[i].v;
            if(dis[x] < INF)
                dis[y] = min(dis[y], dis[x] + edge[i].dist);
        }
    }
    bool flag=1;
  for(int i=0; i<m; ++i){   //判断是否有负环
        int x = edge[i].u, y = edge[i].v;
        if(d[y] > d[x] + edge[i].dist){
            flag = 0; break;
        }
    }
    return flag;
}

SPFA

#include <cstdio>
#include <iostream>
#include <cstring>
#include <queue>
#define maxn 1000005
using namespace std;
const long long INF = 0xffffffff;

int Input(){
    char c;
    for (c = getchar(); c<'0' || c>'9'; c = getchar());
    int a = c - '0';
    for (c = getchar(); c>='0' && c<='9'; c = getchar()) a = a * 10 + c - '0';
    return a;
}

int n, m;
struct edge{
    int e, next, w;
}edge[2][maxn];

long long dis[maxn], ans;
int head[2][maxn], vis[maxn];

inline void spfa(int x){
    for (int i=1; i<=n; i++){
        dis[i] = 0xffffffff;
        //cout << dis[i] << endl;
    }
    memset(vis, 0, sizeof(vis));
    
    queue<int> q;
    int a, b;
    
    q.push(1);
    vis[1] = 1;
    dis[1] = 0;
    
    while(!q.empty()){
        a = q.front(); q.pop();
        vis[a] = 1;
        
        for (int i=head[x][a]; i != -1; i=edge[x][i].next){
            b = edge[x][i].e;
            if (dis[b] > dis[a] + edge[x][i].w){
                dis[b] = dis[a] + edge[x][i].w;
                if (!vis[b]) { q.push(b); vis[b] = 1; }
            }
        }
    }
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 205,236评论 6 478
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 87,867评论 2 381
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 151,715评论 0 340
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,899评论 1 278
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,895评论 5 368
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,733评论 1 283
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 38,085评论 3 399
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,722评论 0 258
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 43,025评论 1 300
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,696评论 2 323
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,816评论 1 333
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,447评论 4 322
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 39,057评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,009评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,254评论 1 260
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 45,204评论 2 352
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,561评论 2 343

推荐阅读更多精彩内容

  • 模板 Dijkstra 模板一(map数组模拟邻接表) 处理小图速度相对较快。内存占用较小,对重边优化较差。 模板...
    染微言阅读 377评论 0 0
  • 简介 搜索迷宫(BFS+队列) 最短路Dijkstra+邻接矩阵Dijkstra+链式前向星+优先队列Bellma...
    染微言阅读 381评论 0 1
  • Android 自定义View的各种姿势1 Activity的显示之ViewRootImpl详解 Activity...
    passiontim阅读 171,478评论 25 707
  • 你就是天使 快乐的天使 你就是魔鬼 可怕的魔鬼 你就是先知 无所不知的神 你就是蟑螂 无处不在的虫 你可以微笑 你...
    布老头和他的家人们阅读 135评论 0 0
  • 时间不等没故事的人,每一个在时光里留下影子记忆的人,必定有一个故事给人们留下印记,就像历史名人。 为了隐私,为了好...
    茗仁阅读 459评论 0 1