Invitation Cards POJ - 1511

题意

求1到其余点的最小距离和与其余点到1点的最小距离和是多少

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
typedef long long ll;
const ll inf=0x3f3f3f3f3f3f3f3f;
const int maxn=1e6+10;
int head[maxn],cnt;
ll d[maxn];
int s[maxn][3];
bool vis[maxn];
int n,m;
struct node
{
    int u,v,w,next;
}p[maxn];
void add(int u,int v,int w)
{
    p[cnt].u=u;
    p[cnt].v=v;
    p[cnt].w=w;
    p[cnt].next=head[u];
    head[u]=cnt++;
}
ll spfa()
{
    for(int i=1;i<=n;i++)
    {
        vis[i]=0;
        d[i]=inf;
    }
    queue<int>q;
    q.push(1);
    vis[1]=1;
    d[1]=0;
    while(!q.empty())
    {
        int t=q.front();
        q.pop();
        vis[t]=0;
        for(int i=head[t];i!=-1;i=p[i].next)
        {
            node e=p[i];
            if(d[e.v]>d[e.u]+e.w)
            {
                d[e.v]=d[e.u]+e.w;
                if(!vis[e.v])
                {
                    vis[e.v]=1;
                    q.push(e.v);
                }
            }
        }
    }
    ll ans=0;
    for(int i=1;i<=n;i++)
        ans=ans+d[i];
    return ans;
}
int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {

        memset(head,-1,sizeof(head));    //初始化
        cnt=0;
        scanf("%d%d",&n,&m);
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&s[i][0],&s[i][1],&s[i][2]);
            add(s[i][0],s[i][1],s[i][2]);
        }
        ll ans=spfa();
        memset(head,-1,sizeof(head));    //初始化
        cnt=0;
        for(int i=0;i<m;i++)
        {
            add(s[i][1],s[i][0],s[i][2]);    //反向添加
        }
        ll temp=spfa();
        printf("%lld\n",ans+temp);
    }
    return 0;
}