POJ 1511 圖論入門第九題

     題目大意：這道題描述非常的蛋疼、英語不好讀了幾遍才懂。就是說求原點到所有點的最短距離+該點到原點的最短距離。

     題目分析：如果不加任何剪枝直接spfa的話會超時，題目給了8000ms但依然不夠。我使用鄰接表來存儲圖、並反向建立表，這樣我們遍歷圖兩遍（正、反）就可以求解出此題。不過這題的建圖我一開始不會。參考了一位大牛的方法、圖的鄰接表寫法真是各式各樣啊，每一種寫法都能解決不同的問題。何時哥才能達到那種境界啊！啊 ? 啊 ?！

     另外注意，這題可能會中間溢出（我定義了unsigned long long），所以對於變量的處理一定要謹慎。

     代碼：

 

#include<cstdio> #include<cstring> #include<queue> #include<algorithm> using namespace std; #define init(a,what) memset(a,what,sizeof(a)) #define read freopen("zx.in","r",stdin) #define write freopen("zx.out","w",stdout) const int INF=0x3f3f3f3f; const int MAXN=1000000+10; typedef unsigned long long int64; typedef struct { int v,next,cost; }ZX;//鄰接表存圖 ZX zx1[MAXN],zx2[MAXN]; int nnum,mnum,ans; int64 sumdist,dist[MAXN],p1[MAXN],p2[MAXN]; void read_G(int a,int b,int c) { //將圖反向建立一遍，可得任何點到原點的最短路（有向圖） ans++; zx1[ans].v=b, zx1[ans].next=p1[a], zx1[ans].cost=c, p1[a]=ans; zx2[ans].v=a, zx2[ans].next=p2[b], zx2[ans].cost=c, p2[b]=ans; } void spfa(int64 *p,ZX e[]) { queue<int>q; bool vis[MAXN]; init(vis,false); while(!q.empty()) q.pop(); for(int i=1;i<=nnum;i++) dist[i]=(i==1 ? 0 : INF); q.push(1); vis[1]=true; while(!q.empty()) { int xx=q.front(); q.pop(); vis[xx]=false; for(int i=p[xx];i!=0;i=e[i].next) { if(dist[e[i].v]>dist[xx]+e[i].cost) { dist[e[i].v]=dist[xx]+e[i].cost; if(!vis[e[i].v]) { vis[e[i].v]=true; q.push(e[i].v); } } } } } int main() { read, write; int ncase; scanf("%d",&ncase); while(ncase--) { ans=0; init(p1,0); init(p2,0); scanf("%d%d",&nnum,&mnum); for(int i=1;i<=mnum;i++) { int a,b,cost; scanf("%d%d%d",&a,&b,&cost); read_G(a,b,cost); } sumdist=0; spfa(p1,zx1);//正向遍歷一遍可求得原點到任意點的最短路 for(int i=1;i<=nnum;i++) sumdist+=dist[i]; spfa(p2,zx2);//反向遍歷求任意點到遠點的最短路 for(int i=1;i<=nnum;i++) sumdist+=dist[i]; printf("%I64d/n",sumdist); } return 0; }