小根堆優化Dijkstra演算法
演算法原理
每次擴充套件一個距離最小的點,再更新與其相鄰的點的距離。
如何尋找距離最小的點
普通的Dijkstra演算法的思路是直接For
i: 1 to n
優化方案是建一個小根堆,小根堆裡儲存由當前結點更新距離的所有點,那麼堆頂就是距離最小的點
如何尋找與源點相鄰的點
當然是鄰接表
具體實現
建一個小根堆heap[]
,用來儲存結點的序號,用一個數組pos[i]
來儲存第i個結點在堆中的位置,用一個標記陣列in_heap[]
來記錄結點是否在堆中,dis[i]
表示到第i個結點的最短距離
對於小根堆的操作還是基本的put()
和get()
,但由於有的結點已經在堆中了,所以可以把put()
完整操作如下:
1.將與源點相鄰的點進行鬆弛操作後加入堆
2.取出位於堆頂的結點
3.若取出的點為終點,則結束演算法
4.將與當前結點相鄰的點進行鬆弛操作
(1)如果該點已經在堆中,就調整在堆中的位置
(2)如果該點不在堆中,就加入堆
5.繼續第二步
例題
最短路徑問題
時間限制:1秒 記憶體限制:256兆
題目描述
平面上有n個點(n<=100),每個點的座標均在-10000~10000之間,其中的一些點之間有連線。若有連線,則表示可從一個點到達另一個點,即兩點間有通路,通路的距離為兩點間的之間距離。現在的任務是找出一點到另一點之間的最短路徑、
輸入
輸入共有n+m+3行,其中:
第一行為整數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<cstdio>
#include <cstring>
#include<cmath>
#define MAXN 100+5
struct Edge//鄰接表
{
int to;
double v;
Edge *next;
};
struct Node
{
int x,y;
}node[MAXN];
int n,m,pos[MAXN],heap_size,s,t,heap[MAXN];
double dis[MAXN];
Edge *first[MAXN];
bool in_heap[MAXN];
void add_edge(int u,int v,double len)
{
Edge *temp=new Edge;
temp->to=v;
temp->v=len;
temp->next=first[u];
first[u]=temp;
}
void calc(int i,int j)
{
double len=sqrt(pow((double)(node[i].x-node[j].x),2)+pow((double)(node[i].y-node[j].y),2));
add_edge(i,j,len);
add_edge(j,i,len);
}
void swapp(int i,int j)
{
int temp=heap[i];
heap[i]=heap[j];
heap[j]=temp;
pos[heap[j]]=j;//調整指標
pos[heap[i]]=i;
}
void shift_up(int now)//調整位置{
int next=0;
while(now>1)
{
next=now>>1;
if(dis[heap[next]]>dis[heap[now]])
swapp(next,now);
now=next;
}
}
void put(int x)//插入堆{
in_heap[x]=true;
heap[++heap_size]=x;
pos[x]=heap_size;
shift_up(heap_size);
}
int get()//取堆頂元素{
int now=1,next,res=heap[1];
in_heap[heap[1]]=false;
heap[1]=heap[heap_size--];
pos[heap[1]]=1;
while(now*2<=heap_size)
{
next=now<<1;
if(next<heap_size&&dis[heap[next+1]]<dis[heap[next]])
++next;
if(heap[now]<=heap[next])
return res;
swapp(now,next);
now=next;
}
return res;
}
void dijkstra()
{
put(s);
dis[s]=0;
while(heap_size>0)
{
int top=get();
if(top==t)
break;
Edge *temp=first[top];
while(temp!=NULL)
{
if(dis[temp->to]>dis[top]+temp->v)
{
dis[temp->to]=dis[top]+temp->v;
//結點在堆中就只調整位置,否則插入堆並調整位置
if(in_heap[temp->to])
shift_up(pos[temp->to]);
else
put(temp->to);
}
temp=temp->next;
}
}
}
int main()
{
int i,x,y;
scanf("%d",&n);
for(i=1;i<=n;++i)
scanf("%d%d",&node[i].x,&node[i].y);
scanf("%d",&m);
for(i=1;i<=m;++i)
{
scanf("%d%d",&x,&y);
calc(x,y);
}
scanf("%d%d",&s,&t);
memset(dis,127,sizeof(dis));
dijkstra();
printf("%.2lf\n",dis[t]);
return 0;
}