pku 1151 Invitation Cards(单源最短路径 SPFA)

 

http://acm.pku.edu.cn/JudgeOnline/problem?id=1511

 

题目大意：在有向图中有n个结点，求从开始结点S（题中默认为1）到各其它各结点的距离和dist1和其它各结点到起点S（1）的距离和dist2之和sum最小.

 

解题思路：这道题点跟边都非常的大，达到1000000，所以就不能用Dj算法啦（因为只会朴素的Dj），就用SPFA来做啦，构建两个邻接表，正向的list1，反向的list2。起点到其他点就用正向邻接表SPFA一次，求总和。之后因为要求其他点到起点的最短路径，肯定不可能做N-1次的SPFA啦，所以就构建反向图，转化成起点到其他点的最短路径和，最后两次结果相加就是答案了。

 

注意：不要用容器(类似vector)做邻接表，动态申请空间耗费太多时间，直接超时（悲剧的我）。后面看了别人的想法才知道用静态链表做邻接表就可以了。

 

#include <stdio.h> #include <queue> #include <iostream> #define Max 1000005 using namespace std; struct node { int adj,w; node *next; }; int P,Q,dist1[Max],dist2[Max],pos; bool visited[Max]; node list1[Max]/*正向图*/,list2[Max]/*反向图*/,mem[2*Max]/*静态空间*/; /*添加邻接点信息，建邻接表*/ void AddAdj(int s,int e,int w,node list[]) { node *p = &mem[pos++]; p->next = NULL; p->adj = e,p->w = w; p->next = list[s].next; list[s].next = p; } void SPFA(int dist[],node List[]) { __int64 sum = 0; node *a; int u,v,w; queue<int>q; q.push(1); visited[1] = 1; dist[1] = 0; while (!q.empty()) { u = q.front(); q.pop(); visited[u] = 0; for (a = List[u].next;a!=NULL;a = a->next) { v = a->adj,w = a->w; if (dist[v]>dist[u]+w) { dist[v] = dist[u]+w; if (!visited[v]) { q.push(v); visited[v] = 1; } } } } } int main() { int test,i,j,w; __int64 sum1,sum2; scanf("%d",&test); while (test--) { scanf("%d%d",&P,&Q); for(i=1;i<=P;i++) visited[i] = 0,dist2[i] = dist1[i] = 0x7fffffff,list1[i].next = list2[i].next = NULL; pos = 0; while (Q--) { scanf("%d%d%d",&i,&j,&w); AddAdj(i,j,w,list1); AddAdj(j,i,w,list2); } SPFA(dist1,list1); for(i=1;i<=P;i++) visited[i] = 0; SPFA(dist2,list2); for(i=1,sum1=sum2=0;i<=P;i++) sum1+=dist1[i],sum2+=dist2[i]; printf("%I64d/n",sum1+sum2); } return 0; }