Dijkstra模板

Dijkstra模板

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector>
using namespace std;
//重点!!!!!
//写整个程序时一定要map[i][j] = map[j][i] 初始化!!!!! 
const int INF=0x3f3f3f3f;
const int maxn=1005;

int dis[maxn],maps[maxn][maxn],n;
bool vis[maxn];

void Init(){
	for(int i = 1;i <= n;i++){
		dis[i] = INF;
		vis[i] = false;
		for(int j = 1;j <= i; j++)
		    i == j ? maps[i][j] = 0 : maps[i][j] = maps[j][i] = INF;
	}
}

void Dijkstra(int s)
{
	/*for(int i = 1; i <= n; i++) 
	    dis[i] = map[s][i];
	vis[s] = true;
	以上三行为一种Dijkstra 起始方式
	另一种:  (替换以上三句即可)*/
    dis[s] = 0;
    
    for(int i=1;i<=n;i++)
    {
        int p=-1,minn=INF;
        for(int j=1;j<=n;j++)
        {
            if(!vis[j]&&dis[j]<minn)
            {
                minn=dis[j];
                p=j;
            }
        }
        vis[p]=1;

        for(int j=1;j<=n;j++)
        {
            if(!vis[j])
            {
                dis[j]=min(dis[j],dis[p]+maps[p][j]);
            }
        }
    }
}

用优先队列优化的Dijkstra 模板

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<algorithm>
#include<stack>
#include<queue>
#include<vector>
#define INF  0x3f3f3f3f 
#defien maxn 1005
using namespace std;
//重点!!!!!
//写整个程序时一定要maps[i][j] = maps[j][i] 初始化!!!!! 

int dis[maxn],maps[maxn][maxn],n,m;


struct Node{
    int n,v;   //存放结点编号和到初始点的距离 
}node;

bool operator < (Node a,Node b){
    a.v == b.v ? a.n>b.n : a.v>b.v;  //先出小 
}

priority_queue<Node> Q;   //优先从小到大

void Init(){
	for(int i = 1;i <= n;i++){
		dis[i] = INF;
		for(int j = 1;j <= i; j++)
		    i == j ? maps[i][j] = 0 : maps[i][j] = maps[j][i] = INF;
	}
}

void Dijkstra(int s){
    while(!Q.empty()) Q.pop();  //清空
	dis[s] = 0;
	Node P;
	P.n = s;
	P.v = 0;
	
	Q.push(P); //将起点放入队列 
	
	while(!Q.empty()){
		
		for(int i = 2; i <= n; i++){
			if(dis[i] > dis[Q.top().n] + maps[Q.top().n][i]){
				dis[i] = dis[Q.top().n] + maps[Q.top().n][i];
				P.n = i;
				P.v = dis[i];
				Q.push(P);
			}
		}
	}
	Q.pop();
}

接下来是邻接表版+优先队列Dijkstra模板

#include <iostream>
#include <cmath>
#include <cstring>
#include <cstdlib>
#include <cstdio>
#include <algorithm>
#include <vector>
#include <queue>
using namespace std;
#define INF 0xfffffff
#define maxn 1002

struct Node
{
    int e;
    int w;
    friend bool operator < (Node A, Node B)
    {
        return  A.w > B.w;
    }
};

bool vis[maxn];

int m, n;
vector< vector<Node> > G;

int Dij(int Star,int End)
{
    Node P, Pn;
    P.e = Star;
    P.w = 0;

    priority_queue<Node> Q;

    Q.push(P);

    while( !Q.empty() )
    {
        P = Q.top();
        Q.pop();

        if( vis[P.e] )
            continue;

        vis[P.e] = true;

        if( P.e == End )
            return P.w;

        int len = G[P.e].size();

        for(int i=0; i< len; i++)
        {
            Pn.e = G[P.e][i].e;
            Pn.w = G[P.e][i].w + P.w;

            if( !vis[Pn.e] )
                Q.push(Pn);
        }
    }
    return -1;
}

int main()
{
    Node P;
    while(cin >> n >> m, m+n)
    {
        G.clear();
        G.resize(n+1);

        memset(vis,false,sizeof(vis));

        for(int i=0; i<m; i++)
        {
            int a, b, c;
            cin >> a >> b >> c;
            P.e = b;
            P.w = c;
            G[a].push_back(P);
            P.e = a;
            G[b].push_back(P);
        }

        int ans = Dij(1,n);

        cout << ans << endl;
    }
    return 0;
}

 

  • 4
    点赞
  • 10
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值