(Til the Cows Come Home )链式前向星存图,优先队列,dijkstra算法

题目链接  USACO 2004 November

下面只是介绍模板,对于这道题还要注意边和顶点的输入顺序,下面的模板都是按照先顶点后边的顺序处理的,直接复制

提交会wa的,所以最好是自己理解着做,要是偷懒直接复制wa了别怪我

常规易于理解的方法
#include <stdio.h>
#include <queue>
#include <string.h>
using namespace std;

// <2> djk求单源最短路,邻接矩阵存图 时间复杂度o(n^2)

const int N = (int) 1000 + 11; // 最大点数
const int INF = (int) 0x3f3f3f3f;

int n, m; // n是点的个数,m是边的个数
int mp[N][N];
void init(int n){// 初始化
	for(int i = 0; i < n; i++){
		for(int j = 0; j < n; j++){
			if(i == j) mp[i][j] = 0;
			else mp[i][j] = mp[j][i] = INF; 
		}
	}
}
void getmap(int m){ // 建图
	int u, v, val;
	while(m--){
		scanf("%d%d%d", &u, &v, &val);
		u--; v--;
		mp[u][v] = mp[v][u] = min(mp[u][v], val); // 如果是双向图写这句
		// mp[u][v] = min(mp[u][v], val); // 如果是单向图写这句
	}
}
bool vis[N]; int dis[N];
void djk(int st, int ed){ // 核心代码,点的序号是从0-(n-1)
	for(int i = 0; i < n; i++){
		vis[i] = false;
		dis[i] = mp[st][i];
	} 
	vis[st] = true;
	for(int i = 1; i < n; i++){
		int mn = INF, id = -1;   //找到离起点最近的未访问过得点
		for(int j = 0; j < n; j++){
			if(!vis[j] && dis[j] < mn) {
				mn = dis[j]; id = j;
			}
		}
		if(id == -1) break;
		vis[id] = true; // 标记
		for(int j = 0; j < n; j++){
			if(!vis[j] && mp[id][j] != INF) { // 如果未访问且 id->
				if(dis[j] > dis[id] + mp[id][j])  // 是否可以松弛j有边
					dis[j] = dis[id] + mp[id][j];
			}
		}
	}
	printf("%d\n", (dis[ed] == INF) ? -1 : dis[ed]);
}
int main(){
	int t; scanf("%d%d", &t, &n);
	init(n);
	getmap(t);
	djk(n - 1, 0);
	return 0;
}

这个是最优化的代码。

//使用链式前向星实现dijkstra,使用链式前向星也可避免图中重边的问题
#include<stdio.h> 
#include<string.h>
#include<algorithm>
#include<queue>
using namespace std;
typedef pair<int,int> pp;
const int maxv = 1005;
const int maxe = 10005;
const int INF = 0x3f3f3f3f;

struct Edge
{
     int next,to,length;
}edge[maxe];
	
int d[maxv],head[maxv];
int v,e,cnt,en;
bool vis[maxv];

void add(int u,int vv,int dis)//链式前向星存图 
{
	edge[++cnt].next = head[u];
	edge[cnt].to = vv;
	edge[cnt].length = dis;
	head[u] = cnt;
}

void dijkstra()
{
	int st = 1;
	memset(vis,0,sizeof(vis));
	memset(d,INF,sizeof(d));
	d[st] = 0;
	priority_queue<pp,vector<pp>,greater<pp> > pa;
	while (!pa.empty())
		pa.pop();
	pa.push(pp(d[st],st));
	while (!pa.empty())
	{
		pp k = pa.top();
		pa.pop();
		if (en==k.second) return;
		if (vis[k.second]) continue;
		vis[k.second] = true;
		for (int i=head[k.second];i!=0;i=edge[i].next)
			if (k.first + edge[i].length < d[edge[i].to])
			{
				d[edge[i].to] = k.first + edge[i].length;
				pa.push(pp(d[edge[i].to],edge[i].to));
			}
	}
}

int main()
{
	cnt = 0;
	scanf("%d %d",&v,&e);//v 是图中的点数,e是图中的边数
	memset(head,0,sizeof(head));
	for (int i=0;i<e;i++)
	{
		int u,vv,dist;
		scanf("%d %d %d",&u,&vv,&dist);
		add(u,vv,dist);
		add(vv,u,dist);//有向图就不用这句了 
	}
	en = v;
	dijkstra();
	printf("%d\n",d[en]);
	return 0;
}

 SPFA模板

#include <stdio.h>
#include <queue>
#include <string.h>
using namespace std;

// <1> spfa求单源最短路,链式前向星存图 时间复杂度o(kE) k是常数,大多数情况下为2

const int N = (int) 1e5 + 11; // 最大点数
const int M = (int) 1e6 + 11; // 最大边数
const int INF = (int) 0x3f3f3f3f;

struct Edge{ // 边的定义
	int to, val, next;
	Edge(){}
	Edge(int _to, int _val, int _next){
		to = _to; val = _val; next = _next;
	}
}edge[M << 1]; // !!如果是双向图的话,边的数量是题目中描述的二倍

int n, m; // n 是图中的点数,m是图中的边数
int head[N], top;
void init(int n){ // 初始化
	memset(head, -1, sizeof(int) * (n + 1));
	top = 0;
}
void Add(int u, int v, int val){ // 加单向边
	edge[top] = Edge(v, val, head[u]);
	head[u] = top++;
}
void getmap(int m){ // 建图
	int u, v, val;
	while(m--){
		scanf("%d%d%d",&u , &v, &val); // u->v有边
		Add(u, v, val);  
		Add(v, u, val); // 如果是双向图,加上这个代码
	}
}

bool vis[N]; int dis[N]; 
void spfa(int st, int ed){ // 核心代码,点序号从0-(n-1)或者1-n都可以
	memset(dis, 0x3f, sizeof(int) * (n + 1));
	memset(vis, false, sizeof(bool) * (n + 1)); // 清空和初始化
	queue<int>que; que.push(st); 
	vis[st] = true; dis[st] = 0;
	while(!que.empty()){
		int u = que.front(); que.pop(); vis[u] = false;
		for(int i = head[u]; ~i; i = edge[i].next){
			Edge e = edge[i];
			if(dis[e.to] > dis[u] + e.val) {//是否可以进行松弛
				dis[e.to] = dis[u] + e.val;
				if(!vis[e.to]){ // 是否在队列中,如果在可以跳过
					que.push(e.to);
					vis[e.to] = true;
				}
			}

		}
	} 
	printf("%d\n", dis[ed] == INF ? -1 : dis[ed]); // st到不了ed输出-1
}

int main(){
	int t; scanf("%d%d", &t, &n);
	init(n);
	getmap(t);
	spfa(n, 1);
	return 0;
}

 floyd模板

 

#include <stdio.h>
#include <queue>
#include <string.h>
using namespace std;

// <4> floyd求多源最短路,邻接矩阵存图 时间复杂度o(n^3),可求得任意两个点之间的最短距离

const int N = (int) 1000 + 11;
const int INF = (int) 0x3f3f3f3f;

int n, m;
int mp[N][N];
void init(int n){ // 初始化
	for(int i = 0; i < n; i++){
		for(int j = 0; j < n; j++){
			if(i == j) mp[i][j] = 0;
			else mp[i][j] = mp[j][i] = INF;
		}
	}
}
void getmap(int m){ // 建图
	int u, v, val;
	while(m--){
		scanf("%d%d%d", &u, &v, &val); u--; v--;
		mp[u][v] = mp[v][u] = min(mp[u][v], val); // 双向图写这行
		//mp[u][v] = min(mp[u][v], val); // 单向图写着行
	}
}
void floyd(int n){ // 核心代码,点的序号为0-n-1
	for(int k = 0; k < n; k++){ 
		for(int i = 0; i < n; i++){
			for(int j = 0; j < n; j++){
				mp[i][j] = min(mp[i][j], mp[i][k] + mp[k][j]); // dp
			}
		}
	}
}

int main(){
	int t; scanf("%d%d", &t, &n);
	init(n);
	getmap(t);
	floyd(n);
	printf("%d\n", mp[n - 1][0] == INF ? -1 : mp[n - 1][0]);
	return 0;
}

 

 

没有更多推荐了,返回首页