#训练5# 2-地铁一日游

题目

https://pintia.cn/problem-sets/994805046380707840/problems/1111914599412858890

思路

啊这，一定要仔细读题目啊，出站有两种可能，一种是计费距离最远（比如3块钱，可以走2+6/6，2+7/6，2+8/6，也就是8，9，10公里，但是只把最远的距离，也就是10公里的站点记录下啦），一种是终点站
所以可以：

1. 用二维数组road存储两点之间的距离，并用Floyd算法更新为最短距离
2. 循环找到每一个站点计费距离最远的站或者终点站，然后用vector记录下来
3. 输入起始站点，dfs一边标记vector里的内容，找到去过那些站
4. 把站排序输出即可
   
   过程图：
   

代码

千辛万苦终于ac惹(* ￣︿￣)(* ￣︿￣)

#include <iostream>
#include <cstring>
#include <algorithm>
#include <queue>
#include <string>
#include <map>

#define maxx 210
#define inf 0x3f3f3f3f
using namespace std;
int road[maxx][maxx];
int dest[maxx];
vector<int> station[maxx];
vector<int> ans;
int n, m, k;
bool vis[maxx];

void floyd()
{
    for (int i = 1; i <= n; ++i)
    {
        for (int j = 1; j <= n; ++j)
        {
            for (int l = 1; l <= n; ++l)
            {
                if (j != l)
                    road[j][l] = min(road[j][i] + road[i][l], road[j][l]);
            }
        }
    }
}

void dfs(int x)
{
    for (int i = 0; i < station[x].size(); ++i)
    {
        int t = station[x][i];
        if (!vis[t])
        {
            vis[t] = 1;
            dfs(t);
        }
    }
}

int main()
{
    //初始化+记录
    cin >> n >> m >> k;
    memset(road, inf, sizeof road);
    while (m--)
    {
        int x, y, l;
        cin >> x;
        dest[x] = 1;  //destination
        while (getchar() != '\n')
        {
            cin >> l >> y;
            if (road[x][y] > l)
                road[x][y] = road[y][x] = l;
            x = y;  //要不然就一直是起点x到各个站点的距离了
        }
        dest[x] = 1; 
    }

    //找到最短的路径
    floyd();

    //计算到每个站点的价格
    for (int i = 1; i <= n; ++i)
    {
        map<int, int> cost;
        for (int j = 1; j <= n; ++j)
        {
            if (road[i][j] != inf)
            {
                int t = 2 + road[i][j] / k;
                cost[t] = max(cost[t], road[i][j]);
            }
        }
        for (int j = 1; j <= n; ++j)
        {
                if (cost[2 + road[i][j] / k] == road[i][j] || i != j && dest[j] != 0&&road[i][j]!=inf)
                {
                    station[i].push_back(j);
                    // cout << i << ' ' << station[i].back() << ' ';
                }
        }
        //   cout<<endl;
    }

    int nn;
    cin >> nn;
    while (nn--)
    {
        int x;
        cin >> x;
        memset(vis, 0, sizeof vis);
        ans.clear();
        vis[x] = 1;
        dfs(x);
        for (int i = 1; i <= n; ++i)
        {
            if (vis[i])
                ans.push_back(i);
        }
        sort(ans.begin(), ans.end());
        ans.erase(unique(ans.begin(), ans.end()), ans.end());  //去掉重复的
        for (int j = 0; j < ans.size(); ++j)
        {
            cout << ans[j];
            if (j != ans.size() - 1)
                cout << ' ';
        }
        cout << endl;
    }
    return 0;
}

总结

1. 全局变量不能过大maxx
   
2. 一开始cost用的是数组，但是后来发现 int t = 2 + road[i][j] / k; 会超界，就改成了map
3. vector里去掉重复元素

使用unique函数

v.erase(unique(v.begin(), v.end()), v.end());
    //unique()函数将重复的元素放到vector的尾部
    //然后返回指向第一个重复元素的迭代器 再用erase函数擦除从这个元素到最后元素的所有的元素

使用set