洛谷 P1342 请柬

Posted 2020-10-03 一蓑烟雨任生平

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，所得dis值相加即为答案。

错因：多定义了三个long long数组。

AC代码：

#include<deque>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 1000010
using namespace std;
int n,m,tot,tot1;
int vis[MAXN],dis[MAXN];
int to[MAXN],cap[MAXN],net[MAXN],head[MAXN];
int to1[MAXN],cap1[MAXN],net1[MAXN],head1[MAXN];
long long ans;
void add(int u,int v,long long w){
    to[++tot]=v;net[tot]=head[u];cap[tot]=w;head[u]=tot;
    to1[++tot1]=u;net1[tot1]=head1[v];cap1[tot1]=w;head1[v]=tot1;
}
void spfa(int s){
    deque<int>que;
    memset(vis,0,sizeof(vis));
    memset(dis,0x3f,sizeof(dis));
    que.push_back(s);
    vis[s]=1;dis[s]=0;
    while(!que.empty()){
        int now=que.front();
        que.pop_front();
        vis[now]=0;
        for(int i=head[now];i;i=net[i]){
            if(dis[to[i]]>dis[now]+cap[i]){
                dis[to[i]]=dis[now]+cap[i];
                if(!vis[to[i]]){
                    vis[to[i]]=1;
                    if(dis[to[i]]>dis[que.front()])
                        que.push_back(to[i]);
                    else
                        que.push_front(to[i]);
                }
            }
        }    
    }
}
void spfa1(int s){
    deque<int>que;
    memset(vis,0,sizeof(vis));
    memset(dis,0x3f,sizeof(dis));
    que.push_back(s);
    vis[s]=1;dis[s]=0;
    while(!que.empty()){
        int now=que.front();
        que.pop_front();
        vis[now]=0;
        for(int i=head1[now];i;i=net1[i]){
            if(dis[to1[i]]>dis[now]+cap1[i]){
                dis[to1[i]]=dis[now]+cap1[i];
                if(!vis[to1[i]]){
                    vis[to1[i]]=1;
                    if(dis[to[i]]>dis[que.front()])
                        que.push_back(to1[i]);
                    else
                        que.push_front(to1[i]);
                }
            }
        }    
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
    }
    spfa(1);
    for(int i=1;i<=n;i++)    ans+=dis[i];
    spfa1(1);
    for(int i=1;i<=n;i++)    ans+=dis[i];
    printf("%lld",ans);
}

TLE代码：

#include<deque>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define MAXN 1000010
using namespace std;
int n,m,tot,tot1;
int vis[MAXN];
int to[MAXN],net[MAXN],head[MAXN];
int to1[MAXN],net1[MAXN],head1[MAXN];
long long dis[MAXN],cap1[MAXN],cap[MAXN],ans;
void add(int u,int v,long long w){
    to[++tot]=v;net[tot]=head[u];cap[tot]=w;head[u]=tot;
    to1[++tot1]=u;net1[tot1]=head1[v];cap1[tot1]=w;head1[v]=tot1;
}
void spfa(int s){
    deque<int>que;
    memset(vis,0,sizeof(vis));
    memset(dis,0x3f,sizeof(dis));
    que.push_back(s);
    vis[s]=1;dis[s]=0;
    while(!que.empty()){
        int now=que.front();
        que.pop_front();
        vis[now]=0;
        for(int i=head[now];i;i=net[i]){
            if(dis[to[i]]>dis[now]+cap[i]){
                dis[to[i]]=dis[now]+cap[i];
                if(!vis[to[i]]){
                    vis[to[i]]=1;
                    if(dis[to[i]]>dis[que.front()])
                        que.push_back(to[i]);
                    else
                        que.push_front(to[i]);
                }
            }
        }    
    }
}
void spfa1(int s){
    deque<int>que;
    memset(vis,0,sizeof(vis));
    memset(dis,0x3f,sizeof(dis));
    que.push_back(s);
    vis[s]=1;dis[s]=0;
    while(!que.empty()){
        int now=que.front();
        que.pop_front();
        vis[now]=0;
        for(int i=head1[now];i;i=net1[i]){
            if(dis[to1[i]]>dis[now]+cap1[i]){
                dis[to1[i]]=dis[now]+cap1[i];
                if(!vis[to1[i]]){
                    vis[to1[i]]=1;
                    if(dis[to[i]]>dis[que.front()])
                        que.push_back(to1[i]);
                    else
                        que.push_front(to1[i]);
                }
            }
        }    
    }
}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++){
        int u,v,w;
        scanf("%d%d%d",&u,&v,&w);
        add(u,v,w);
    }
    spfa(1);
    for(int i=1;i<=n;i++)    ans+=dis[i];
    spfa1(1);
    for(int i=1;i<=n;i++)    ans+=dis[i];
    printf("%lld",ans);
}