http://poj.org/problem?id=1511

题目大意:给出n个点和n条有向边,求所有点到源点1的来回最短路之和(保证每个点都可以往返源点1)

单源点来回最短距离问题,可以调用一次迪杰斯特拉之后将地图反过来再调用一次迪杰斯特拉,两个结果之和就是结果。

但是,此题数据过大,用普通的迪杰斯特拉模板绝对是不行的,需要用到迪杰斯特拉的优先队列实现的模板。

最近才学到的用优先队列优化后的迪杰斯特拉算法。。

先贴模板吧。

 

//Asimple  HDU  1874
#include<bits/stdc++.h>
#define INF 0xfffffff
using namespace std;
const int maxn = 1005;
struct node {
	int to;
	int w;
	node(){
	}
	node(int to, int w) : to(to), w(w) {
	}
	bool operator < (const node& a) const {
		return w>a.w;
	}
};
vector<node> e[maxn];
int n, m;
int dis[maxn];
void init() {
	for(int i=0; i<maxn; i++) {
		e[i].clear();
		dis[i] = INF;
	}
}

void Dijkstra(int st, int ed) {
	priority_queue<node> q;
	dis[st] = 0;
	q.push(node(st,dis[st]));
	while( !q.empty() ) {
		node now = q.top();
		q.pop();
		for(int i=0; i<e[now.to].size(); i++) {
			node y = e[now.to][i];
			if( dis[y.to]>now.w+y.w) {
				dis[y.to] = now.w + y.w;
				q.push(node(y.to, dis[y.to]));
			}
		}
	}
	if( dis[ed]==INF) printf("-1\n");
	else printf("%d\n", dis[ed]);
}


int main() {
	while( cin >> n >> m ) {
		init();
		for(int i=0; i<m; i++) {
			int x, y, num;
			scanf("%d %d %d", &x, &y, &num);
			e[x].push_back(node(y,num));
			e[y].push_back(node(x,num));
		}
		int s, t;
		scanf("%d%d",&s, &t);
		Dijkstra(s, t);
	}
	return 0;
}

 

 做的时候错了十几遍。实在受不了了。百度了大佬们的题解。发现原来是INF设小了。。

//Asimple
//#include <bits/stdc++.h>
#include <iostream>
#include <sstream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <vector>
#include <cctype>
#include <cstdlib>
#include <stack>
#include <cmath>
#include <set>
#include <map>
#include <string>
#include <queue>
#include <limits.h>
#include <time.h>
#define INF 1e10
#define mod 999101
#define PI 3.14159265358979323
#define swap(a,b,t) t = a, a = b, b = t
#define CLS(a, v) memset(a, v, sizeof(a))
#define debug(a)  cout << #a << " = "  << a <<endl
using namespace std;
typedef long long ll;
const int maxn = 1000005;
int n, m, num, T, k, len, ans, sum, x, y;
ll dis[maxn];
int fa[maxn];
bool vis[maxn];
struct node{ int y; int v; node( int y, int v) : y(y), v(v) { } node (){ } bool operator < (const node &a) const { return dis[y]>dis[a.y]; } }; struct eg{ int now; int w; int next; }; eg e[maxn]; int a[maxn][3]; void adde(int y, int x, int w) { e[len].now = x; e[len].w = w; e[len].next = fa[y]; fa[y] = len++; } void init() { len = 0; for(int i=0; i<=n; i++) { fa[i] = -1; dis[i] = INF; vis[i] = false; } } ll solve() { priority_queue<node> q; dis[1] = 0; q.push(node(1, dis[1])); while( !q.empty() ) { node pp = q.top(); q.pop(); if( vis[pp.y] ) continue; vis[pp.y] = true; for(int i=fa[pp.y]; i!=-1; i=e[i].next){ int yy = e[i].now; int cost = e[i].w; if( dis[yy]>dis[pp.y]+cost){ dis[yy] = dis[pp.y]+cost; q.push(node(yy,dis[yy])); } } } ll res = 0; for(int i=1; i<=n; i++) { res += dis[i]; } return res; } void input() { for(scanf("%d", &T); T--; ){ scanf("%d %d", &n, &m); init(); for(int i=0; i<m; i++) { scanf("%d%d%d", &a[i][0], &a[i][1], &a[i][2]); adde(a[i][0], a[i][1], a[i][2]); } ll cnt = solve(); init(); CLS(e, 0); for(int i=0; i<m; i++) adde(a[i][1], a[i][0], a[i][2]); cnt += solve(); printf("%lld\n", cnt); } } int main(){ input(); return 0; }