标签:
对于给出具有非负权重的边和源顶点S的图G,算法可在O(mlogn)时间内找出从s点到其他每一个顶点的距离。
如果图是稠密的,即对于某个ε>0,m>=n^(1+ε),可以被改善在O(m/ε)内执行。(m为图的边数,n为图的定点数)
1 struct HeapElement { 2 // key存储顶点序号,value存储到该顶点的最短距离 3 int key, value; 4 }; 5 struct MinHeap { 6 HeapElement heap[MAXN]; 7 int position[MAXN]; 8 int size; // 顶点数 9 void init() { 10 heap[size=0].value = -INF; 11 memset(position, 0, sizeof(position)); 12 } 13 void insert(int key, int value) { 14 heap[++size].key = key; 15 heap[size].value = value; 16 position[key] = size; 17 siftUp(size); 18 } 19 void decrease(int index) { 20 int x = heap[index].value; 21 int y = heap[size].value; 22 -- size; 23 if (index == size+1) 24 return; 25 26 swap(heap[index], heap[size+1]); 27 if (y >= x) { 28 siftDown(index); 29 } else { 30 siftUp(index); 31 } 32 } 33 int delmin() { 34 int x = heap[1].key; 35 decrease(1); 36 return x; 37 } 38 void siftUp(int index) { 39 while (index > 1) { 40 if (heap[index].value < heap[index/2].value) { 41 swap(heap[index],heap[index/2]); 42 } else { 43 break; 44 } 45 index /= 2; 46 } 47 } 48 void siftDown(int index) { 49 while (index*2 <= size) { 50 index *= 2; 51 if (index < size && heap[index].value > heap[index+1].value) { 52 ++ index; 53 } 54 if (heap[index].value < heap[index/2].value) { 55 swap(heap[index],heap[index/2]); 56 } else { 57 break; 58 } 59 } 60 } 61 void makeHeap() { 62 for (int i = size/2; i > 0; -- i) 63 siftDown(i); 64 } 65 void swap(HeapElement &a, HeapElement &b) { 66 HeapElement temp = a; 67 a = b; 68 b = temp; 69 int tmp = position[a.key]; 70 position[a.key] = position[b.key]; 71 position[b.key] = tmp; 72 } 73 }H;
#include <iostream> #define INF 0x7FFFFFFF using namespace std; const int SIZE = 105; int dist[SIZE]; int G[SIZE][SIZE]; bool vis[SIZE]; struct HeapElement { int key, value; }; void swap(HeapElement &ha, HeapElement &hb) { int key = ha.key; int value = ha.value; ha.key = hb.key; ha.value = hb.value; hb.key = key; hb.value = value; }; // 使用邻接表储存图,线性表储存堆 struct MinHeap { HeapElement heap[SIZE]; int n; // 顶点数 void makeheap() { for (int i = n/2; i > 0; -- i) siftDown(i); }; void siftUp(int index) { int k = index; while (k > 1) { if (heap[k].value < heap[k/2].value) { swap(heap[k],heap[k/2]); } else { break; } k /= 2; } }; void siftDown(int index) { int k = index; while (k*2 <= n) { k *= 2; if (k < n && heap[k].value > heap[k+1].value) { ++ k; } if (heap[k].value < heap[k/2].value) { swap(heap[k],heap[k/2]); } else { break; } } }; void insert(HeapElement element) { heap[++n] = element; siftUp(n); }; void decrease(int index) { int x = heap[index].value; int y = heap[n].value; n -= 1; // 若删除节点位于最末位置,则删除成功,无需其他操作。 if (index == n+1) return; heap[index] = heap[n+1]; if (y >= x) { siftDown(index); } else { siftUp(index); } }; int decreaseMin() { int x = heap[1].key; decrease(1); return x; }; }H; void dijkstra(int src, int n) { int i, j, w; bool flag; for (i = 1; i <= n; ++ i) { if (G[i][src] != INF) { dist[i] = G[src][i]; HeapElement h = {i, dist[i]}; H.insert(h); } else { dist[i] = INF; } } memset(vis, false, sizeof(vis)); vis[src] = true; dist[src] = 0; for (i = 1; i < n; ++ i) { int node = H.decreaseMin(); vis[node] = true; for (w = 1; w <= n; ++ w) { flag = false; if (!vis[w] && G[node][w] != INF) { if (dist[node] < dist[w] - G[node][w]) { dist[w] = dist[node] + G[node][w]; } for (j = 1; j <= H.n; ++ j) { if (H.heap[j].key == w) { H.heap[j].value = dist[w]; flag = true; break; } } if (!flag) { HeapElement h = {w, dist[w]}; H.insert(h); } else { H.siftUp(j); } } } } }; void init(int n) { for (int i = 1; i <= n; ++ i) for (int j = 1; j <= n; ++ j) G[i][j] = INF; H.n = 0; }; int main() { int N, M, a, b, c; //freopen("C:\\Users\\Smile\\test.txt","r",stdin); //freopen("C:\\Users\\Smile\\out.txt", "w", stdout); while (scanf("%d%d",&N,&M)!=EOF, N&&M) { init(N); for (int i = 0; i < M; ++ i) { scanf("%d%d%d",&a,&b,&c); if (G[a][b] > c) { G[a][b] = c; G[b][a] = c; } } dijkstra(1, N); printf("%d\n",dist[N]); } }
标签:
原文地址:http://www.cnblogs.com/SSYYGAM/p/5434518.html