POJ1511(Invitation Cards)

最短路的题，这题时间限制较严，第一次用dijkstra果断挂掉，后来改用SPFA写，由于我用的数据结构是动态链表，所以还是TLE，改为静态链表后就AC了。在这个过程中还贡献了一次WA，因为最后结果用32位会溢出，题中说的总价格不会超过1000000000是单个最短路不会超，而最后的结果是n-1个结点的最短路之和。

View Code

 1 #include <stdio.h>
 2 #include <stdlib.h>
 3 #include <memory.h>
 4 #define INF 1000000001
 5 #define N 1000001
 6 #define MIN(a,b) ((a)<(b)?(a):(b))
 7 struct node
 8 {
 9     int v,w;
10     struct node *next;
11 };
12 int cnt;
13 struct node edge[2*N];
14 struct node *a[N],*b[N];
15 int dist[N],n,m;
16 int queue[N],front,rear;
17 char vis[N];
18 void addEdge(int u,int v,int w)
19 {
20     struct node *p,*q;
21     p=&edge[cnt++];
22     q=&edge[cnt++];
23     p->v=v,p->w=w;
24     p->next=a[u],a[u]=p;
25     q->v=u,q->w=w;
26     q->next=b[v],b[v]=q;
27 }
28 void SPFA(int u,struct node *list[])
29 {
30     int i,v;
31     struct node *p;
32     front=rear=0;
33     memset(vis,0,sizeof(vis));
34     for(i=0;i<n;i++)    dist[i]=(i==u?0:INF);
35     queue[rear++]=u;
36     while(front!=rear)
37     {
38         i=queue[front],front=(front+1)%N;
39         vis[i]=0;
40         for(p=list[i];p;p=p->next)
41         {
42             v=p->v;
43             if(dist[v]>dist[i]+p->w)
44             {
45                 dist[v]=dist[i]+p->w;
46                 if(!vis[v])
47                 {
48                     vis[v]=1;
49                     queue[rear]=v,rear=(rear+1)%N;
50                 }
51             }
52         }
53     }
54 }
55 int main()
56 {
57     int i,j,k,d,t;
58     long long ans;
59     scanf("%d",&t);
60     while(t--)
61     {
62         scanf("%d%d",&n,&m);
63         cnt=0;
64         for(i=0;i<m;i++)
65         {
66             scanf("%d%d%d",&j,&k,&d);
67             j--,k--;
68             addEdge(j,k,d);
69         }
70         ans=0;
71         SPFA(0,a);
72         for(i=1;i<n;i++)   ans+=dist[i];
73         SPFA(0,b);
74         for(i=1;i<n;i++)   ans+=dist[i];
75         printf("%lld\n",ans);
76         memset(a,0,sizeof(a));
77         memset(b,0,sizeof(b));
78     }
79     return 0;
80 }