PAT L3-011. 直捣黄龙

最短路。

先求出一个只包含最短路的$DAG$，然后在$DAG$上$dp$一下就可以了。英文变数字还有$map$转一下。

#include<map>
#include<set>
#include<ctime>
#include<cmath>
#include<queue>
#include<string>
#include<stack>
#include<vector>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<functional>
using namespace std;

int n,m,s,t,num[];
char tmp[],A[],B[];

map<string,int>xx;
map<int,string>yy;

vector<int>G[],nG[];
int Time[][];

int disS[],disT[],f[];

int U[*],V[*],In[];

int ss[],pre[],sl[],dp[];

void spfaS()
{
    memset(f,,sizeof f);
    for(int i=;i<=n;i++) disS[i]=0x7FFFFFFF;
    queue<int>Q; Q.push(s); f[s]=; disS[s]=;

    while(!Q.empty())
    {
        int h = Q.front(); Q.pop(); f[h]=;
        for(int i=;i<G[h].size();i++)
        {
            int to = G[h][i];
            if(disS[h]+Time[h][to]<disS[to])
            {
                disS[to] = disS[h]+Time[h][to];
                if(f[to]==)
                {
                    f[to]=;
                    Q.push(to);
                }
            }
        }
    }
}

void spfaT()
{
    memset(f,,sizeof f);
    for(int i=;i<=n;i++) disT[i]=0x7FFFFFFF;
    queue<int>Q; Q.push(t); f[t]=; disT[t]=;

    while(!Q.empty())
    {
        int h = Q.front(); Q.pop(); f[h]=;
        for(int i=;i<G[h].size();i++)
        {
            int to = G[h][i];
            if(disT[h]+Time[h][to]<disT[to])
            {
                disT[to] = disT[h]+Time[h][to];
                if(f[to]==)
                {
                    f[to]=;
                    Q.push(to);
                }
            }
        }
    }
}

void W()
{
    queue<int>Q; Q.push(s); dp[s]=;

    while(!Q.empty())
    {
        int h = Q.front(); Q.pop();
        for(int i=;i<nG[h].size();i++)
        {
            int to = nG[h][i];
            dp[to]+=dp[h];
            if(sl[h]+>sl[to])
            {
                sl[to] = sl[h]+;
                ss[to] = ss[h]+num[to];
                pre[to]=h;
            }

            else if(sl[h]+==sl[to])
            {
                if(ss[h]+num[to]>ss[to])
                {
                    sl[to] = sl[h]+;
                    ss[to] = ss[h]+num[to];
                    pre[to]=h;
                }
            }
            In[to]--;
            if(In[to]==) Q.push(to);
        }
    }
}

int main()
{
    scanf("%d%d",&n,&m); scanf("%s%s",A,B);

    xx[A]=; yy[]=A;
    for(int i=;i<=n;i++)
    {
        scanf("%s",tmp); xx[tmp]=i; yy[i]=tmp;
        scanf("%d",&num[i]);
    }
    s=; t=xx[B];

    for(int i=;i<=m;i++)
    {
        int P;
        scanf("%s%s",A,B); scanf("%d",&P);
        int u = xx[A], v = xx[B];

        U[i] = u; V[i]=v;
        Time[u][v]=Time[v][u]=P;
        G[u].push_back(v);
        G[v].push_back(u);
    }

    spfaS(); spfaT();

    for(int i=;i<=m;i++)
    {
        if(disS[U[i]]+disT[V[i]]+Time[U[i]][V[i]]==disS[t])
        {
            nG[U[i]].push_back(V[i]);
            In[V[i]]++;
        }
        if(disS[V[i]]+disT[U[i]]+Time[U[i]][V[i]]==disS[t])
        {
            nG[V[i]].push_back(U[i]);
            In[U[i]]++;
        }
    }

    W();

    vector<int>Ans;
    stack<int>St;

    int now=t;
    while()
    {
        if(now==) break;
        St.push(now);
        now=pre[now];
    }

    while(!St.empty())
    {
        Ans.push_back(St.top());
        St.pop();
    }

    for(int i=;i<Ans.size();i++)
    {
        cout<<yy[Ans[i]];
        if(i<Ans.size()-) printf("->");
        else printf("\n");
    }

    printf("%d %d %d\n",dp[t],disS[t],ss[t]);

    return ;
}