POJ 1511 Invitation Cards

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

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

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

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

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

先贴模板吧。

 

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61

//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;
}