AtCoder Beginner Contest 335 E题 Non-Decreasing Colorful Path

最新推荐文章于 2024-11-07 18:59:45 发布

超哥聊信奥

最新推荐文章于 2024-11-07 18:59:45 发布

阅读量975

点赞数 7

分类专栏：题目题解文章标签：算法 c++ 数据结构图论学习

本文链接：https://blog.csdn.net/oitutor/article/details/138343438

版权

题目题解专栏收录该内容

96 篇文章 2 订阅

订阅专栏

E题：Non-Decreasing Colorful Path

标签：最短路、 $d ijk s t r a$
题意：给定一个 $n$ 个顶点和 $m$ 条边的无向图，每个顶点上有分数 $a_i$ ，求从顶点 $1$ 到顶点 $n$ 得分最高的路径。得分是路径中顶点分数不同的顶点数目，要保证路径上的顶点分数是不递减的（包含等于）
比如路径上顶点的分数分别为 $10$ $20$ $20$ $30$ $40$ => 那么得分为 $4$
题解：比较典型的 $d ijk s t r a$ 的变型题（加上约束条件），因为题目要求不递减，堆优化部分可以按每个顶点的分数从小到大排序，松弛部分操作的时候保证是从低分数的顶点到高分数（相等也进入更新）的顶点，跑最长路；需要注意处理等于的时候，得分是顶点中分数不同的顶点数目。
代码：

#include <bits/stdc++.h>
using namespace std;

typedef long long ll;
ll a[200005], d[200005];
vector<ll> e[200005];
priority_queue< pair<ll, ll> > q;

int main() {
    ll n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i++) cin >> a[i];
    for (int i = 1; i <= m; i++) {
        ll u, v;
        cin >> u >> v;
        e[u].push_back(v);
        e[v].push_back(u);
    }

    d[1] = 1;
    q.push(make_pair(-a[1], 1));

    while (!q.empty()) {
        int u = q.top().second;
        q.pop();
        for (auto v: e[u]) {
            if (a[v] >= a[u]) {
                if (d[v] < d[u] + (a[v] != a[u])) {
                    d[v] = d[u] + (a[v] != a[u]);
                    q.push(make_pair(-a[v], v));
                }
            }
        }
    }

    cout << d[n];
    return 0;
}