PAT (Advanced Level) 1018. Public Bike Management (30)

先找出可能在最短路上的边，图变成了一个DAG，然后在新图上DFS求答案就可以了。

#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<cstdio>
#include<queue>
#include<vector>
using namespace std;
 
const int INF=0x7FFFFFFF;
const int maxn=;
int n,m,C1,C2,tot;
int Cmax;
int val[maxn];
struct Edge
{
    int u,v,L;
}e[];
int f[];
vector<int>g[maxn];
int dis[][maxn];
 
int ans[maxn],cnt;
int sta[maxn];
 
int MIN1,MIN2;
 
void init()
{
    MIN1=MIN2=INF;
    tot=;C1=;
    for(int i=;i<=;i++) g[i].clear();
}
 
void add(int a,int b,int c)
{
    e[tot].u=a,e[tot].v=b,e[tot].L=c;
    g[a].push_back(tot);
    tot++;
}
 
void read()
{
    scanf("%d%d%d%d",&Cmax,&n,&C2,&m);
    for(int i=;i<=n;i++) scanf("%d",&val[i]);
    for(int i=;i<=m;i++)
    {
        int u,v,L; scanf("%d%d%d",&u,&v,&L);
        add(u,v,L);
        add(v,u,L);
    }
}
 
void SPFA(int f,int st)
{
    for(int i=;i<=;i++) dis[f][i]=INF;
    dis[f][st]=;
    int flag[maxn]; memset(flag,,sizeof flag);
    queue<int>Q; Q.push(st); flag[st]=;
    while(!Q.empty())
    {
        int head= Q.front(); Q.pop(); flag[head]=;
        for(int i=;i<g[head].size();i++)
        {
            int id=g[head][i];
            if(dis[f][head]+e[id].L<dis[f][e[id].v])
            {
                dis[f][e[id].v]=dis[f][head]+e[id].L;
                if(flag[e[id].v]==)
                {
                    Q.push(e[id].v);
                    flag[e[id].v]=;
                }
            }
        }
    }
}
 
void dfs(int x,int num1,int num2,int deep)
{
    sta[deep]=x;
 
    if(x==C2)
    {
        if(num2<MIN2)
        {
            MIN2=num2;
            MIN1=num1;
            for(int i=;i<=deep;i++) ans[i]=sta[i];
            tot=deep;
        }
 
        else if(num2==MIN2&&num1<MIN1)
        {
            MIN2=num2;
            MIN1=num1;
            for(int i=;i<=deep;i++) ans[i]=sta[i];
            tot=deep;
        }
        return;
    }
 
    for(int i=;i<g[x].size();i++)
    {
        if(f[g[x][i]]==) continue;
        int to=e[g[x][i]].v;
 
        if(val[to]==Cmax/) dfs(to,num1,num2,deep+);
        else if(val[to]>Cmax/) dfs(to,num1+val[to]-Cmax/,num2,deep+);
        else if(val[to]<Cmax/)
        {
            if(num1>Cmax/-val[to]) dfs(to,num1+val[to]-Cmax/,num2,deep+);
            else dfs(to,,num2+Cmax/-val[to]-num1,deep+);
        }
    }
}
 
void work()
{
    int len=dis[][C2];
 
    for(int i=;i<tot;i++)
        if(dis[][e[i].u]+e[i].L+dis[][e[i].v]==len)
            f[i]=;
 
    dfs(,,,);
 
    printf("%d ",MIN2);
    for(int i=;i<=tot;i++)
    {
        if(i<tot) printf("%d->",ans[i]);
        else printf("%d ",ans[i]);
    }
    printf("%d\n",MIN1);
}
 
int main()
{
    init();
    read();
    SPFA(,C1);
    SPFA(,C2);
    work();
    return ;
}