[HDU 1535]Invitation Cards[SPFA反向思维]

题意: (欧洲人自己写的题面就是不一样啊...各种吐槽...果断还是看晕了)

有向图, 有个源叫CCS, 求从CCS到其他所有点的最短路之和, 以及从其他所有点到CCS的最短路之和.

思路:

返回的时候是多个源,但是因为终点只有一个,所以把所有边反向之后, 再SPFA一次源即可.

#include<cstdio>
#include<vector>
#include<queue>
const int MAXN=1000000+10;
typedef long long ll;
const ll inf=1e60;
using namespace std;
struct Node{
    int v,w;
};
vector<Node>mp1[MAXN];//正向建图
vector<Node>mp2[MAXN];//反向建图
int n,m;
ll cost[MAXN];
 
void SPFA(int u,vector<Node>mp[]){
    for(int i=2;i<=n;i++)cost[i]=inf;
    cost[1]=0;
    queue<int>Q;
    Q.push(u);
    while(!Q.empty()){
        int u=Q.front();
        Q.pop();
        for(int i=0;i<mp[u].size();i++){
            int v=mp[u][i].v;
            int w=mp[u][i].w;
            if(cost[v]>cost[u]+w){
                cost[v]=cost[u]+w;
                Q.push(v);
            }
        }
    }
}
 
int main(){
    int _case;
    scanf("%d",&_case);
    while(_case--){
        scanf("%d%d",&n,&m);
        for(int i=1;i<=n;i++){
            mp1[i].clear();
            mp2[i].clear();
        }
        for(int i=1;i<=m;i++){
            int u,v,w;
            scanf("%d%d%d",&u,&v,&w);
            Node p1,p2;
            p1.v=u,p2.v=v,p1.w=p2.w=w;
            mp1[u].push_back(p2);
            mp2[v].push_back(p1);
        }
        SPFA(1,mp1);//正向求一次
        ll ans=0;
        for(int i=2;i<=n;i++){
            ans+=cost[i];
        }
        SPFA(1,mp2);//反向求一次
        for(int i=2;i<=n;i++){
            ans+=cost[i];
        }
        printf("%lld\n",ans);
    }
    return 0;
}

自己敲一遍:

#include <cstdio>
#include <vector>
#include <cstring>
#include <queue>
using namespace std;
typedef long long ll;
const int MAXN = 1e6+5;
const ll INF = 0x3f3f3f3f3f3f3f3fll;
typedef struct node
{
    int v,w;
    node(){}
    node(int _v, int _w):v(_v),w(_w){}
}node;
 
int n,m;
bool inq[MAXN];
ll cost[MAXN];
vector<node> g1[MAXN],g2[MAXN];
 
ll SPFA(int op)
{
    memset(cost,0x3f,sizeof(cost));
    memset(inq,false,sizeof(inq));
    cost[1] = 0;
    queue<int> q;
    q.push(1);
    inq[1] = true;
    while(!q.empty())
    {
        int now = q.front();q.pop();
        inq[now] = false;
        for(int i=0,v,w;i<((op==1)?g1[now].size():g2[now].size());i++)
        {
            if(op==1)
            {
                v = g1[now][i].v, w = g1[now][i].w;
            }
            else
            {
                v = g2[now][i].v, w = g2[now][i].w;
            }
            if(cost[v] > cost[now] + w)
            {
                cost[v] = cost[now] + w;
                if(!inq[v])
                {
                    q.push(v);
                    inq[v] = true;
                }
            }
        }
    }
    ll ret = 0;
    for(int i=2;i<=n;i++)
        ret += cost[i];
    return ret;
}
 
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d %d",&n,&m);
        for(int i=1;i<=n;i++)
        {
            g1[i].clear();
            g2[i].clear();
        }
        for(int i=0,u,v,w;i<m;i++)
        {
            scanf("%d %d %d",&u,&v,&w);
            g1[u].push_back(node(v,w));
            g2[v].push_back(node(u,w));
        }
        ll ans = 0;
        ans += SPFA(1);
        ans += SPFA(2);
        printf("%d\n",(int)ans);
    }
}