HDU 1535 Invitation Cards 正反向建图

转载请注明该文链接

题意：有n个车站（n为10^6范围），第一个车站是出发点。现在要从第一个车站安排志愿者去2-n号车站服务，每个车站一人。志愿者早上坐公交车从一号车站出发，晚上从各车站坐公交车回到一号车站。每条线路都有各自的费用。求最小总开销。（假设所有车站一定能到达，每次需要出发或者返回时公交车只载志愿者中的一人）

思路：有两种思路：

1.以1号车站为源点求SPFA，然后把从1号车站到各公交车站的最小花费累加。再分别以2-n号车站为源点，求出他们以1号车站为终点的最小花费并累加，所有累加值即为最小总开销。但是由于题目范围很大，所以这种思路会TLE。

2.注意到1中以2-n号车站为源点时均以1号车站为终点，而这步可以通过反向建图实现，即把所有u->v变成v->u，然后以1号车站为源点，可以一次性得到所有结果，而且复杂度也降低了。于是这种做法需要建两幅图，一幅正向，一幅反向，分别以1号车站为源点求SPFA，然后把2-n号车站的两个图的结果全部相加就可以得到最小总开销。

注意要用64位长整型，注意车站是从1号到n号。

这里SPFA可以用SLF优化，时间可以变快。关于SPFA的优化方法可以看这里。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<queue>
#include<algorithm>

using namespace std;

typedef __int64 ll;

const int MAXV=1000010;
const int MAXE=1000010;
const ll inf=1LL<<60;

struct node
{
    int v;
    ll dis;
    node *next;
}E[MAXE<<1],*G1[MAXV],*G2[MAXV],*head;

inline void add(int u,int v,ll dis,node *G[])
{
    head->v=v;
    head->dis=dis;
    head->next=G[u];
    G[u]=head++;
}

int n,m;
ll d[MAXV];
bool inq[MAXV];

void init()
{
    memset(inq,false,sizeof(inq));
    memset(G1,0,sizeof(G1));
    memset(G2,0,sizeof(G2));
    head=E;
}

void SPFA(int s,ll d[],node *G[])
{
    deque<int> Q;
    Q.push_front(s);
    fill(d,d+MAXV,inf);
    d[s]=0;
    while(!Q.empty())
    {
        int u=Q.front();
        Q.pop_front();
        inq[u]=false;
        for(node *p=G[u];p;p=p->next)
        {
            int v=p->v;
            ll dis=p->dis;
            if(d[u]+dis<d[v])
            {
                d[v]=d[u]+dis;
                if(!inq[v])
                {
                    inq[v]=true;
                    if(!Q.empty() && d[v]<=d[Q.front()]) Q.push_front(v);
                    else Q.push_back(v);
                }
            }
        }
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        init();
        scanf("%d%d",&n,&m);
        int u,v;
        ll dis;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%I64d",&u,&v,&dis);
            add(u,v,dis,G1);
            add(v,u,dis,G2);
        }
        ll ans=0;
        SPFA(1,d,G1);
        for(int i=2;i<=n;i++) ans+=d[i];
        SPFA(1,d,G2);
        for(int i=2;i<=n;i++) ans+=d[i];
        printf("%I64d\n",ans);
    }
    return 0;
}