PAT 1018 Public Bike Management (30 分)

```#include <cstdio>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
const int inf=99999999;
//g为邻接表，weight存储每个节点的单车数量，d表示从出发点到各点的最短距离
int g[501][501],weight[501],d[501];
int n,cmax,sp,m,numpath=0,minneed=inf,minremain=inf;
bool vis[501]={false};
//pre[i]表示第[i]个节点的前驱
vector<int> pre[501];
//path存储路径，temppath存储临时路径
vector<int> path,temppath;
void dijkstra(int s)
{
    fill(d,d+501,inf);
    d[s]=0;
    for(int i=0;i<=n;i++)
    {
        int u=-1,MIN=inf;
        for(int j=0;j<=n;j++)
        {
            if(vis[j]==false&&d[j]<MIN)
            {
                u=j;
                MIN=d[j];
            }
        }
        if(u==-1)
            return;
        vis[u]=true;
        for(int v=0;v<=n;v++)
        {
            if(vis[v]==false&&g[u][v]!=inf)
            {
                //有更短的路径，清空pre[v]
                if(d[u]+g[u][v]<d[v])
                {
                    d[v]=d[u]+g[u][v];
                    pre[v].clear();
                    pre[v].push_back(u);
                }
                //最短路径长度相同，入队
                else if(d[u]+g[u][v]==d[v])
                    pre[v].push_back(u);
            }
        }
    }
}
void DFS(int v)
{
    //当前节点为出发节点，临界条件
    if(v==0)
    {
        temppath.push_back(v);
        //need表示需要带出去的自行车辆数，remain表示需要带回去的车辆数
        int need=0,remain=0;
        //遍历路径
        for(int i=temppath.size()-1;i>=0;i--)
        {
            int id=temppath[i];
            //节点车多于一半，带走
            if(weight[id]>0)
                remain+=weight[id];
            else
            {
                //节点车少于一半且从前面的节点搜集的车量数大于缺额
                if(remain>abs(weight[id])) remain-=abs(weight[id]);
                //从前面的节点搜集的车量数小于等于缺额
                else
                {
                    need+=abs(weight[id])-remain;
                    remain=0;
                }
            }
        }
        //根据题意更新路径
        if(need<minneed)
        {
            minneed=need;
            minremain=remain;
            path=temppath;
        }
        else if(need==minneed&&minremain>remain)
        {
            minremain=remain;
            path=temppath;
        }
        temppath.pop_back();
        return;
    }
    temppath.push_back(v);
    for(int i=0;i<pre[v].size();i++)
        DFS(pre[v][i]);
    temppath.pop_back();
}
int main()
{
    fill(g[0],g[0]+501*501,inf);
    scanf("%d%d%d%d",&cmax,&n,&sp,&m);
    //每个节点的车辆数直接减最大容量一半，之后直接用正负来判断需要补充还是清理
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&weight[i]);
        weight[i]-=cmax/2;
    }
    int u,v;
    for(int i=0;i<m;i++)
    {
        scanf("%d%d",&u,&v);
        scanf("%d",&g[u][v]);
        g[v][u]=g[u][v];
    }
    dijkstra(0);
    DFS(sp);
    printf("%d ",minneed);
    for(int i=path.size()-1;i>=0;i--)
    {
        printf("%d",path[i]);
        if(i>0)
            printf("->");
    }
    printf(" %d",minremain);
    return 0;
}