浙大 PAT 甲级 1018 Public Bike Management C++ 要点说明及代码实现

分析

用单源最短路径似乎不太合适，因为题目要求考虑多条最短路径时的选择问题，选择的依据是需要从中心拿出的单车数量和拿回中心的单车数量。看到这里，解题方法应该很清晰明了，最容易想到的就是深度优先搜索，这跟前面有道题很类似，而且深度优先搜索的编码大同小异，建议多找几道类似的题目集中做，提高DFS的编码速度。

下面说一下可能会出现的错误。 如果你已经写完了深度优先所搜的代码却有少数几个用例没有通过时，请继续看。

我感觉这题题意表达的不是很清晰，有一些地方只能凭借猜测：
调整过程。如果认为来回的路上都会调整，也就是说，仅仅关注这一路上站点已有车辆的总和，来计算出需要携带或送回的单车数，那么会有两个用例无法通过。根据题意推测，应该只有去的时候调整沿途的站点，回来的时候不会调整。因此，可能会出现既有需要携带单车过去、最后又回拿回一些单车的情况。

比如每个站点最大容量为10,5是最佳数量。从0到3会依次经过1、2、3三个站点，这三个站点的车辆数为2,6,10.那么，由于经过站点1时需要补充，因此需要携带3辆单车，经过2和3时会冗余，需要带回1+5=6辆单车。所以输出是3 0->1->2->3 6。

而如果认为来回的路上都会调整，那么得出只需要带回3辆的结论，因为2+6+10-3*5=3.

另外题目对于多条最短路径时的选择也没有表述清楚。在题目最开始的表述为：

 If there are more than one shortest path, the one that requires the least number of bikes sent from PBMC will be chosen.

但是在OUTPUT的说明中呢，又说了一句：

Note that if such a path is not unique, output the one that requires minimum number of bikes that we must take back to PBMC. The judge's data guarantee that such a path is unique. 

所以姑且认为选择的优先级是路径长度>送出的单车数>送回的单车数。

这题就这么愉快的解决啦！这题的测试用例还比较友善，只要过程弄清楚了就行。

 

代码

#include<stdio.h>
#include<vector>
#include<math.h>
using namespace std;

struct E
{
    int des;
    int cost;
};

int condition[501];
bool mark[501];
vector<E> edges[501];
vector<int> bestRoute;
int bestBring = 100000;
int bestBack = 100000;
int minSum = 100000;
int C;
void DFS(int sta, int des, int sum, vector<int> route, int bring, int back)
{
    for (int i = 0; i < edges[sta].size(); i++)
    {
        if (mark[edges[sta][i].des] == true)
        {
            continue;
        }
        int tmpbring = bring;
        int tmpback = back;
        if (condition[edges[sta][i].des] < C)
        {
            if (tmpback >= C - condition[edges[sta][i].des])
            {
                tmpback -= C - condition[edges[sta][i].des];
            }
            else
            {
                tmpbring += C - condition[edges[sta][i].des] - tmpback;
                tmpback = 0;
            }
        }
        else if (condition[edges[sta][i].des] > C)
        {
            tmpback += condition[edges[sta][i].des] - C;
        }
        sum += edges[sta][i].cost;
        route.push_back(edges[sta][i].des);
        mark[edges[sta][i].des] = true;
        if (edges[sta][i].des == des)
        {
            if (sum < minSum || (sum == minSum&&tmpbring < bestBring) || (sum == minSum&&tmpbring == bestBring&&tmpback < bestBack))
            {
                minSum = sum;
                bestBring = tmpbring;
                bestBack = tmpback;
                bestRoute.clear();
                bestRoute.assign(route.begin(), route.end());
            }
        }
        else
        {
            DFS(edges[sta][i].des, des, sum, route, tmpbring, tmpback);
        }
        sum -= edges[sta][i].cost;
        route.pop_back();
        mark[edges[sta][i].des] = false;
    }
    return;
}
int main()
{
    vector<int> route;
    route.clear();
    for (int i = 0; i < 501; i++)
    {
        edges[i].clear();
        mark[i] = false;
    }
    int N, S, M;
    scanf("%d%d%d%d", &C, &N, &S, &M);
    C /= 2;
    for (int i = 1; i <= N; i++)
    {
        scanf("%d", &condition[i]);
    }
    for (int i = 0; i < M; i++)
    {
        int sta, des, cost;
        scanf("%d%d%d", &sta, &des, &cost);
        E e;
        e.des = des;
        e.cost = cost;
        edges[sta].push_back(e);
        e.des = sta;
        edges[des].push_back(e);
    }
    mark[0] = true;
    DFS(0, S, 0, route, 0, 0);
    printf("%d 0", bestBring);

    for (int i = 0; i < bestRoute.size(); i++)
    {
        printf("->%d", bestRoute[i]);
    }

    printf(" %d", bestBack);
    return 0;
}