洛谷P1629 邮递员送信

题目链接

        邮递员送信 - 洛谷

题目描述

        有一个邮递员要送东西，邮局在节点 1。他总共要送 n-1 样东西，其目的地分别是节点 2 到节点 n。由于这个城市的交通比较繁忙，因此所有的道路都是单行的，共有 m 条道路。这个邮递员每次只能带一样东西，并且运送每件物品过后必须返回邮局。求送完这 n-1 样东西并且最终回到邮局最少需要的时间。

输入格式

        第一行包括两个整数，n 和 m，表示城市的节点数量和道路数量。

        第二行到第 (m+1)行，每行三个整数，u,v,w 表示从 u 到 v 有一条通过时间为 w 的道路。

输出格式

        输出仅一行，包含一个整数，为最少需要的时间。

输入样例

5 10
2 3 5
1 5 5
3 5 6
1 2 8
1 3 8
5 3 4
4 1 8
4 5 3
3 5 6
5 4 2

输出样例

83

思路：

        将问题拆解为两部分。

        1.邮递员从邮局到每一个结点的距离：是一个标准的单源最短路径问题，用Dijkstra算法解决，看一下数据范围，m<nlongn,该图算是稀疏图，可以用二叉堆加速。

        2.计算每一个结点到邮局的距离，是一个多对一的问题，可以将每个结点都用Dijkstra跑一遍，然后得出该节点到邮局的距离，这样的时间复杂度为O(n(m+n)logn),可能会爆。

        所以采用将距离矩阵整体转置，转置后，原先从u到v的距离，就变为了从v到u的距离，此时变将问题从多对一转化为一对多，此时再用dijkstra算法跑一遍即可获得每个点到邮局的距离。

#include<iostream>
#include<queue>
#include<cstring>
using namespace std;

const int MAXN=1005;
const int MAXM=100005;
#define INFINITY 0X7f7f7f7f

struct node
{
    int dis;
    int pos;
    bool operator<(const node a)const{
        return a.dis<dis;
    }
};

struct Edge
{
    int to;
    int w;
    int next;
};

Edge e[MAXM];
int head[MAXN],dis[MAXN],cnt=0;
bool vis[MAXN];
int n,m;
priority_queue<node>q;
int uvw[MAXM][3];

void addEdge(int u,int v,int w)
{
    cnt++;
    e[cnt].to=v;
    e[cnt].w=w;
    e[cnt].next=head[u];
    head[u]=cnt;
}

void Dijkstra(int s)
{
    dis[s]=0;
    q.push(node{0,s});

    while(!q.empty())
    {
        node tmp=q.top();
        q.pop();

        if(vis[tmp.pos]) continue;
        vis[tmp.pos]=true;

        for(int i=head[tmp.pos];i;i=e[i].next)
        {
            int y=e[i].to;
            if(dis[y]>dis[tmp.pos]+e[i].w)
            {
                dis[y]=dis[tmp.pos]+e[i].w;
            }
            if(!vis[y])
            {
                q.push(node{dis[y],y});
            }
        }
    }
}

int main()
{
    //freopen("C:\\Users\\lenovo\\Desktop\\P1629_1.in","r",stdin);
    //freopen("C:\\Users\\lenovo\\Desktop\\ans.txt","w",stdout);
    memset(dis,INFINITY,sizeof dis);
    memset(head,0,sizeof head);
    memset(vis,false,sizeof vis);
    long long ans=0;
    cin>>n>>m;
    for(int i=1;i<=m;++i)
    {
        cin>>uvw[i][0]>>uvw[i][1]>>uvw[i][2];
        addEdge(uvw[i][0],uvw[i][1],uvw[i][2]);
    }

    Dijkstra(1);

    for(int i=2;i<=n;++i)
    {
        ans+=dis[i];
    }

    cnt=0;
    memset(dis,INFINITY,sizeof dis);
    memset(head,0,sizeof head);
    memset(vis,false,sizeof vis);
    for(int i=1;i<=m;++i)
    {
        addEdge(uvw[i][1],uvw[i][0],uvw[i][2]);
    }

    Dijkstra(1);

    for(int i=2;i<=n;++i)
    {
        ans+=dis[i];
    }

    cout<<ans<<endl;

    return 0;
}