luogu P1342 请柬

Posted 2021-02-17 charlesss

dij板子的灵活运用

在一个有向图中求：1.1号点到剩下所有点的最短路 2.剩下所有点到一号点的最短路

1相信谁都会，存有向图跑一遍

2是这道题的精髓，应用一个比较有价值的思想：在有向图中跑反向边，相当于求出所有点到一号点的最短路

所以简单说就是存两个版本的图然后两遍dij板子

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define ll long long
#define re register
#define il inline
#define inf 214748364
#define fp(i,a,b) for(re int i=a;i<=b;i++)
#define fq(i,a,b) for(re int i=a;i>=b;i--)
using namespace std;
int n,m,h[1000005],cnt=0,inq[1000005]={},dis[1000005]={},h1[1000005]={};
ll ans=0;
struct Edge
{
    int to,next,w;
}e[1000005],e1[1000005];
struct Node
{
    int l,d;
    Node(){}
    Node(int l,int d):l(l),d(d){}
    bool operator < (const Node& zsy) const
        {
            return l>zsy.l;
        }
};
il void add(re int u,re int v,re int w)
{
    e[++cnt]=(Edge){v,h[u],w};h[u]=cnt;
}
il void add1(re int u,re int v,re int w)
{
    e1[cnt]=(Edge){v,h1[u],w};h1[u]=cnt;
}
il void Dijkstra()
{
    fp(i,1,n) dis[i]=inf;
    dis[1]=0;
    priority_queue<Node>Q;
    Q.push(Node(0,1));
    while(!Q.empty())
    {
        Node u=Q.top();Q.pop();
        if(inq[u.d]) continue;
        inq[u.d]=1;
        for(re int i=h[u.d];i+1;i=e[i].next)
        {
            re int v=e[i].to,w=e[i].w;
            if(u.l+w<dis[v])
            {
                dis[v]=u.l+w;
                Q.push(Node(dis[v],v));
            }
        }
    }
}
il void Dijkstra1()
{
    fp(i,1,n) dis[i]=inf;
    dis[1]=0;
    memset(inq,0,sizeof(inq));
    priority_queue<Node>Q;
    Q.push(Node(0,1));
    while(!Q.empty())
    {
        Node u=Q.top();Q.pop();
        if(inq[u.d]) continue;
        inq[u.d]=1;
        for(re int i=h1[u.d];i+1;i=e1[i].next)
        {
            re int v=e1[i].to,w=e1[i].w;
            if(u.l+w<dis[v])
            {
                dis[v]=u.l+w;
                Q.push(Node(dis[v],v));
            }
        }
    }
}
int main()
{
    memset(h,-1,sizeof(h));memset(h1,-1,sizeof(h1));
    scanf("%lld%lld",&n,&m);
    fp(i,1,m)
    {
        ll u,v,w;
        scanf("%lld%lld%lld",&u,&v,&w);
        add(u,v,w);
        add1(v,u,w);
    }
    Dijkstra();
    fp(i,2,n) ans+=dis[i];
    Dijkstra1();
    fp(i,2,n) ans+=dis[i];
    printf("%lld
",ans);
    return 0;
}