牛客网-2018 美团 CodeM 编程大赛-资格赛-5-你的城市

ACM模版

描述

牛客网-2018 美团 CodeM 编程大赛-资格赛-5-你的城市_第1张图片
牛客网-2018 美团 CodeM 编程大赛-资格赛-5-你的城市_第2张图片
牛客网-2018 美团 CodeM 编程大赛-资格赛-5-你的城市_第3张图片
牛客网-2018 美团 CodeM 编程大赛-资格赛-5-你的城市_第4张图片

题解

这个题十分有意思,题目要求要选择的路线必须在保证无论哪班车错过后都有补救措施的前提条件下代价最低。

这样,也就意味着,我们找到的路代价最低,且要求该路线方案每个节点(可以除去终点)都必须有两条及其以上的方案可以通往终点。

所以首先我们需要预处理出来每个节点能否到达终点,这里也就存在一个问题,这里有时间限制,为了规避这个限制,我们需要进行拆点,因为保证都是整点或者半点的时间限制,所以我们可以将每个城市都拆分成 49 49 个点,然后再添加一个源点一个汇点,这样就可以顺利预处理出来每个点能否到达终点。

接着,我们需要考虑路线方案中所过城市,必须保证在该时间节点半个小时候依然有方案可以到达终点,所以我们可以在 Dijkstra D i j k s t r a 算法求最短路的过程中,过滤掉不符合这个要求的节点(这个节点只是某一时刻往后该城市是不符合条件的,因为城市根据时间拆分为 49 49 个节点),这样就可以保证最后找到的路线方案无论错过哪班车肯定存在补救方案。

这里需要注意的是,换车需要半小时的时间消耗、错过车也需要再半个小时的时间消耗,你也可以理解他去重新买票了,所以需要多半个小时,蛤蛤蛤。

具体的看代码吧,虽然不好想,但是代码很容易理解。

代码

#include 
#include 
#include 
#include 

using namespace std;

const int INF = 0x3f3f3f3f;
const int MAXN = 5e5 + 10;
const int MAGIC = 49;

struct node
{
    int v, c;

    node(int v_ = 0, int c_ = 0) : v(v_), c(c_) {}

    bool operator < (const node &r)const
    {
        return c > r.c;
    }
};

struct edge
{
    int v, cost;

    edge(int v_ = 0, int cost_ = 0) : v(v_), cost(cost_) {}
};

int n, m;
int dis[MAXN];
int cnt[MAXN];
bool vis[MAXN];
vector E[MAXN];

void Dijkstra(int n, int start)
{
    memset(vis, false, sizeof(vis));
    memset(dis, 0x3f, sizeof(dis));

    priority_queue pqn;

    dis[start] = 0;
    pqn.push(node(start, 0));

    node tmp;
    while (!pqn.empty())
    {
        tmp = pqn.top();
        pqn.pop();

        int u = tmp.v;
        //  保证有补救措施
        if (!cnt[u + 1])
        {
            continue;
        }

        if (vis[u])
        {
            continue;
        }

        vis[u] = true;
        for (int i = 0; i < E[u].size(); i++)
        {
            int v = E[tmp.v][i].v;
            int cost = E[u][i].cost;
            if (!vis[v] && dis[v] > dis[u] + cost)
            {
                dis[v] = dis[u] + cost;
                pqn.push(node(v, dis[v]));
            }
        }
    }
}

void addedge(int u, int v, int w)
{
    E[u].push_back(edge(v, w));
}

bool dfs(int u)
{
    if (u == n * MAGIC + 2)
    {
        return cnt[u] = 1;
    }

    int flag = 0;
    size_t sz = E[u].size();
    for (int i = 0; i < sz; i++)
    {
        int v = E[u][i].v;
        if (!vis[v])
        {
            vis[v] = 1;
            if (dfs(v))
            {
                flag = 1;
            }
        }
        else
        {
            if (cnt[v])
            {
                flag = 1;
            }
        }
    }

    return cnt[u] = flag;
}

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

    int x, y, c, ts_h, ts_m, td_h, td_m;
    for (int i = 1; i <= m; i++)
    {
        scanf("%d%d%d%d:%d%d:%d", &x, &y, &c, &ts_h, &ts_m, &td_h, &td_m);

        //  拆点,半小时一个点,先拆做 MAGIC 个点
        int ts_id = ts_h * 2 + (ts_m / 30) + 1;
        int td_id = td_h * 2 + (td_m / 30) + 1;
        //  如果某站直达终点则不需要考虑延时半小时
        for (int j = td_id + (y == n ? 0 : 1); j <= MAGIC; j++)
        {
            addedge((x - 1) * MAGIC + ts_id, (y - 1) * MAGIC + j, c);
        }
    }

    //  n * MAGIC + 1 号点为源点,在任何时刻由源点到达北京代价为 0
    //  n * MAGIC + 2 号点为汇点,在任何时刻由小城到达汇点代价为 0
    for (int i = 1; i <= MAGIC; i++)
    {
        addedge(n * MAGIC + 1, i, 0);
        addedge((n - 1) * MAGIC + i, n * MAGIC + 2, 0);
    }

    //  从源点开始搜索
    vis[n * MAGIC + 1] = 1;
    dfs(n * MAGIC + 1);

    for (int i = 1; i <= n; i++)
    {
        for (int j = MAGIC; j >= 1; j--)
        {
            cnt[(i - 1) * MAGIC + j] += cnt[(i - 1) * MAGIC + j + 1];
        }
    }

    memset(vis, 0, sizeof(vis));
    Dijkstra(n * MAGIC + 2, n * MAGIC + 1);

    if (dis[n * MAGIC + 2] == INF)
    {
        cout << "-1\n";
    }
    else
    {
        cout << dis[n * MAGIC + 2] << "\n";
    }

    return 0;
}

你可能感兴趣的:(dfs,&&,bfs,图论)