洛谷 P1342 请柬

题目描述

在电视时代,没有多少人观看戏剧表演。Malidinesia古董喜剧演员意识到这一事实，他们想宣传剧院,尤其是古色古香的喜剧片。他们已经打印请帖和所有必要的信息和计划。许多学生被雇来分发这些请柬。每个学生志愿者被指定一个确切的公共汽车站，他或她将留在那里一整天,邀请人们参与。

这里的公交系统是非常特殊的:所有的线路都是单向的，连接两个站点。公共汽车离开起始点，到达目的地之后又空车返回起始点。学生每天早上从总部出发，乘公交车到一个预定的站点邀请乘客。每个站点都被安排了一名学生。在一天结束的时候,所有的学生都回到总部。现在需要知道的是，学生所需的公交费用的总和最小是多少。

输入输出格式

输入格式：

 

第1行有两个整数n、m(1<=n,m<=1000000)，n是站点的个数，m是线路的个数。

然后有m行，每行描述一个线路，包括3个整数，起始点，目的地和价格。

总部在第1个站点，价钱都是整数，且小于1000000000。

 

输出格式：

 

输出一行，表示最小费用。

 

输入输出样例

输入样例#1：

4 6
1 2 10
2 1 60
1 3 20
3 4 10
2 4 5
4 1 50

输出样例#1：

210 

说明

【注意】

此题数据规模较大，需要使用较为高效的算法，此题不设小规模数据分数。

思路：正、反边SPFA

#include<cstdio>
#include<cstring>
#include<queue>
#include<iostream>
#define MAX_M 1000010
using namespace std;
int n,m,head[MAX_M],hed[MAX_M],tot1,tot2,dis[MAX_M];
long long ans=0;
struct Edge {
    int from,to,next,value;
}e[MAX_M];
bool exist[MAX_M];
queue<int>q;
struct Edge2 {
    int from,to,next,value;
}ee[MAX_M];// 反边 hed tot2
void Add_Edge1(int u,int v,int w){
    e[++tot1].from=u;e[tot1].to=v;e[tot1].value=w;
    e[tot1].next=head[u];head[u]=tot1;
}
void Add_Edge2(int u,int v,int w){
    ee[++tot2].from=u;ee[tot2].to=v;ee[tot2].value=w;
    ee[tot2].next=hed[u];hed[u]=tot2;
}
void prepare(){
    memset(exist,false,sizeof(exist));
    memset(dis,0x3f,sizeof(dis));
    while(!q.empty())q.pop();
}
void SPFA1(){
    exist[1]=true;q.push(1);dis[1]=0;
    while(!q.empty()){
        int p=q.front();exist[p]=false;q.pop();
        for(int i=head[p];i;i=e[i].next){
            int v=e[i].to,w=e[i].value;
            if(dis[v]>dis[p]+w){
                dis[v]=dis[p]+w;
                if(!exist[v]){
                    q.push(v);exist[v]=true;
                }
            }
        }
    }
}
void SPFA2(){
    exist[1]=true;q.push(1);dis[1]=0;
    while(!q.empty()){
        int p=q.front();exist[p]=false;q.pop();
        for(int i=hed[p];i;i=ee[i].next){
            int v=ee[i].to,w=ee[i].value;
            if(dis[v]>dis[p]+w){
                dis[v]=dis[p]+w;
                if(!exist[v]){
                    q.push(v);exist[v]=true;
                }
            }
        }
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1,u,v,w;i<=m;i++){
        scanf("%d%d%d",&u,&v,&w);
        Add_Edge1(u,v,w);Add_Edge2(v,u,w);
    }
    prepare();
    SPFA1();
    for(int i=1;i<=n;i++) ans+=dis[i];
    prepare();
    SPFA2();
    for(int i=1;i<=n;i++) ans+=dis[i];
    printf("%lld",ans);
    return 0;
}