POJ1511(Invitation Cards)

题目链接

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

#include <stdio.h>
#include <stdlib.h>
#include <memory.h>
#define INF 1000000001
#define N 1000001
#define MIN(a,b) ((a)<(b)?(a):(b))
struct node
{
    int v,w;
    struct node *next;
};
int cnt;
struct node edge[2*N];
struct node *a[N],*b[N];
int dist[N],n,m;
int queue[N],front,rear;
char vis[N];
void addEdge(int u,int v,int w)
{
    struct node *p,*q;
    p=&edge[cnt++];
    q=&edge[cnt++];
    p->v=v,p->w=w;
    p->next=a[u],a[u]=p;
    q->v=u,q->w=w;
    q->next=b[v],b[v]=q;
}
void SPFA(int u,struct node *list[])
{
    int i,v;
    struct node *p;
    front=rear=0;
    memset(vis,0,sizeof(vis));
    for(i=0;i<n;i++)    dist[i]=(i==u?0:INF);
    queue[rear++]=u;
    while(front!=rear)
    {
        i=queue[front],front=(front+1)%N;
        vis[i]=0;
        for(p=list[i];p;p=p->next)
        {
            v=p->v;
            if(dist[v]>dist[i]+p->w)
            {
                dist[v]=dist[i]+p->w;
                if(!vis[v])
                {
                    vis[v]=1;
                    queue[rear]=v,rear=(rear+1)%N;
                }
            }
        }
    }
}
int main()
{
    int i,j,k,d,t;
    long long ans;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m);
        cnt=0;
        for(i=0;i<m;i++)
        {
            scanf("%d%d%d",&j,&k,&d);
            j--,k--;
            addEdge(j,k,d);
        }
        ans=0;
        SPFA(0,a);
        for(i=1;i<n;i++)   ans+=dist[i];
        SPFA(0,b);
        for(i=1;i<n;i++)   ans+=dist[i];
        printf("%lld\n",ans);
        memset(a,0,sizeof(a));
        memset(b,0,sizeof(b));
    }
    return 0;
}