最短路 + 二分答案 - 奥格瑞玛

洛谷 - 1462 通往奥格瑞玛的道路

题目背景

在艾泽拉斯大陆上有一位名叫歪嘴哦的神奇术士，他是部落的中坚力量

有一天他醒来后发现自己居然到了联盟的主城暴风城

在被众多联盟的士兵攻击后，他决定逃回自己的家乡奥格瑞玛

题目描述

在艾泽拉斯，有n个城市。编号为1,2,3,...,n。

城市之间有m条双向的公路，连接着两个城市，从某个城市到另一个城市，会遭到联盟的攻击，进而损失一定的血量。

没经过一个城市，都会被收取一定的过路费（包括起点和终点）。路上并没有收费站。

假设1为暴风城，n为奥格瑞玛，而他的血量最多为b，出发时他的血量是满的。

歪嘴哦不希望花很多钱，他想知道，在可以到达奥格瑞玛的情况下，他所经过的所有城市中最多的一次收取的费用的最小值是多少。

输入输出格式

输入格式：

第一行3个正整数，n，m，b。分别表示有n个城市，m条公路，歪嘴哦的血量为b。

接下来有n行，每行1个正整数，fi。表示经过城市i，需要交费fi元。

再接下来有m行，每行3个正整数，ai，bi，ci(1<=ai，bi<=n)。表示城市ai和城市bi之间有一条公路，如果从城市ai到城市bi，或者从城市bi到城市ai，会损失ci的血量。

输出格式：

仅一个整数，表示歪嘴哦交费最多的一次的最小值。

如果他无法到达奥格瑞玛，输出AFK。

说明

对于60%的数据，满足n≤200，m≤10000，b≤200

对于100%的数据，满足n≤10000，m≤50000，b≤1000000000

对于100%的数据，满足ci≤1000000000，fi≤1000000000，可能有两条边连接着相同的城市。

一、题目分析

这道题是一道一眼图论最短路的题，但是题目中多出了一个限制 —— 通过每个城市要收费，也就是对于节点的限制。题目中要求询问在 dist[n - 1] < HP 的情况下，路径权值最大节点权值的最小值。

首先分析题目，题目要求的是一个典型的最大值最小的问题，很容易想到二分答案。

二、解题思路

现在考察我们要二分的是什么东西，价值的限制是对于节点的，我们可以二分这个当前的最大值，然后在SPFA求最短路的过程中，如果遍历到的点的权值大于这个限制，就不让这个点加入路径。

我们如果得到了答案就更新一次 ans ，最终如果一次都没更新，那必然是 AFK。

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
struct edge
{
    int vertex,weight;
    edge(int vv,int ww)
    {
        vertex = vv;
        weight = ww;
    }
};
int n,m,b; // vertex - N, edge - M , hp - b;
int ans = 0x3f3f3f3f;
int city[10005]; // 经过城市花钱
int city_nosort[10005];
int in_queue[10005];
int dist[10005]; // 到 i 的最短路
vector<vector<edge> > edges(10005);
void init()
{
    memset(city_nosort,0,sizeof(city_nosort));
    memset(city,0,sizeof(city));
    memset(in_queue,0,sizeof(in_queue));
}
void spfa(int val)
{
    if (city_nosort[0] > val) return ; // 源点就不满足条件 ，直接退出
    memset(dist,0x3f,sizeof(dist));
    queue<int> q;
    q.push(0);
    in_queue[0] = true;
    dist[0] = 0;
    while(!q.empty())
    {
        int temp = q.front();
        q.pop();
        in_queue[temp] = false;
        for (int i=0;i<edges[temp].size();i++)
        {
            int &vtx = edges[temp][i].vertex;
            if (city_nosort[vtx] > val) continue; //当前节点的权值大于限制，直接进下一个点
            if (dist[temp] + edges[temp][i].weight < dist[vtx])
            {
                dist[vtx] = dist[temp] + edges[temp][i].weight;
                if (in_queue[vtx] == false)
                {
                    q.push(vtx);
                    in_queue[vtx] = true;
                }
            }
        }
    }
}


int main()
{
    init();
    cin >> n >> m >> b;
    for(int i=0;i<n;i++)
    {
        scanf("%d",&city[i]); // i 城市  city[i] 的过路费
        city_nosort[i] = city[i];
    }
    for (int i=0;i<m;i++)
    {
        int fm,to,weight;
        scanf("%d%d%d",&fm,&to,&weight);
        fm--;to--;
        edges[fm].push_back(edge(to,weight));
        edges[to].push_back(edge(fm,weight));
    }
    sort(city,city+n);
    int l=0,r=n-1;
    while(l <= r)
    {
        int mid = (l + r) / 2;
        spfa(city[mid]);
        if (b - dist[n-1] <= 0)
        {
            l = mid + 1;
        }
        else
        {
            r = mid - 1;
            ans = city[mid];
        }
    }
    if (ans == 0x3f3f3f3f)
        cout << "AFK";
    else
        cout << ans;
    return 0;
}