POJ 1511 - Invitation Cards (dijkstra优先队列)

题目链接:http://poj.org/problem?id=1511

就是求从起点到其他点的最短距离加上其他点到起点的最短距离的和 , 注意路是单向的.

因为点和边很多, 所以用dijkstra优先队列的做法.

起点到其他点的最短距离之和就是dij一下 . 要求其他点到起点的最短距离的话就是把原先边的方向反向一下,然后再求起点到其他点的最短距离之和 , 同样dij一下.

代码如下:

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <vector>
 #include <queue>

 using namespace std;
 const int MAXN = 1e6 + ;
 typedef long long LL;
 typedef pair <LL , LL> P;
 vector <P> edge[MAXN]; //这里邻接表二元组(pair)的second代表边的终点标号, first代表边的距离
 bool ok[MAXN];
 LL d[MAXN] , INF = 1e12;
 int a[MAXN] , b[MAXN] , c[MAXN];  //分别代表边的起点 终点 权值

 void init(int n) {
     for(int i =  ; i <= n ; i++) {
         d[i] = INF;
         ok[i] = false;
         edge[i].clear();
     }
 }

 void dijkstra(int s) {
     d[s] = ;
     priority_queue <P , vector<P> , greater<P> > Q;  //优先队列 二元组的first:最短距离 second:点的标号
     Q.push(P( , s));   //压入起点
     while(!Q.empty()) {
         P p = Q.top();
         Q.pop();
         int v = p.second;
         if(ok[v]) {
             continue;
         }
         ok[v] = true;
         for(int i =  ; i < edge[v].size() ; i++) {
             P temp = edge[v][i];
             if(d[temp.second] > d[v] + temp.first) {
                 d[temp.second] = d[v] + temp.first;
                 Q.push(P(d[temp.second] , temp.second));
             }
         }
     }
 }

 int main()
 {
     int t , n , m;
     LL u , v , cost;
     scanf("%d" , &t);
     while(t--) {
         scanf("%d %d" , &n , &m);
         init(n);
         for(int i =  ; i < m ; i++) {
             scanf("%d %d %d" , &a[i] , &b[i] , &c[i]);
             edge[a[i]].push_back(P(LL(c[i]) , LL(b[i])));
         }
         LL res = ;
         dijkstra();
         for(int i =  ; i <= n ; i++) {
             res += d[i];
         }
         init(n);
         for(int i =  ; i < m ; i++) {
             edge[b[i]].push_back(P(LL(c[i]) , LL(a[i])));   //每条边反向一下
         }
         dijkstra();
         for(int i =  ; i <= n ; i++) {
             res += d[i];
         }
         printf("%lld\n" , res);
     }
 }