結果

問題 No.1607 Kth Maximum Card
ユーザー hir355hir355
提出日時 2021-07-16 22:51:07
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2,725 ms / 3,500 ms
コード長 2,787 bytes
コンパイル時間 2,854 ms
コンパイル使用メモリ 211,112 KB
最終ジャッジ日時 2025-01-23 02:30:24
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 33
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <atcoder/modint>
#include <bits/stdc++.h>

using namespace std;
using namespace atcoder;

constexpr long long INF_LL = 2000000000000000000LL;
constexpr int INF = 2000000000;
constexpr long long MOD = 1000000007;

#define all(x) x.begin(), x.end()
#define REP(i, a, b) for(int i = a; i < b; i++)
#define rep(i, n) REP(i, 0, n)

typedef long long ll;
typedef pair<ll, ll> P;
typedef vector<int> vi;
typedef vector<vi> vvi;
typedef vector<P> vp;
typedef vector<ll> vl;

int dx[4] = {0, -1, 0, 1};
int dy[4] = {1, 0, -1, 0};
int sign[2] = {1, -1};

template <class T> bool chmax(T &a, T b) {
    if(a < b) {
        a = b;
        return 1;
    }
    return 0;
}
template <class T> bool chmin(T &a, T b) {
    if(a > b) {
        a = b;
        return 1;
    }
    return 0;
}

ll modpow(ll a, ll b, ll m) {
    if(b == 0)
        return 1;
    ll t = modpow(a, b / 2, m);
    if(b & 1) {
        return (t * t % m) * a % m;
    } else {
        return t * t % m;
    }
}

struct edge {
    int to;
    ll cost;
    edge(int t, ll c) { to = t, cost = c; }
};

typedef vector<vector<edge>> graph;

using mint = modint998244353;

vector<ll> dijkstra(vector<vector<edge>> &g, int start, int n) {
    priority_queue<P, vector<P>, greater<P>> que;
    vector<ll> dist(n, INF_LL);
    dist[start] = 0;
    que.push(P(0, start));
    while(!que.empty()) {
        P p = que.top();
        que.pop();
        int v = p.second;
        if(dist[v] < p.first)
            continue;
        rep(i, (int)g[v].size()) {
            edge e = g[v][i];
            if(dist[e.to] > dist[v] + e.cost) {
                dist[e.to] = dist[v] + e.cost;
                que.push(P(dist[e.to], e.to));
            }
        }
    }
    return dist;
}

int main(){
    cin.tie(0);
    std::ios_base::sync_with_stdio(false);
    std::cout << std::fixed << std::setprecision(16);

    int n, m, k;
    cin >> n >> m >> k;
    vector<pair<pair<int, int>, int>> e(m);
    rep(i, m){
        cin >> e[i].first.first >> e[i].first.second >> e[i].second;
    }
    sort(all(e), [](auto &x, auto &y){return x.second < y.second;});
    int l = -1, r = 200001;
    while(r - l > 1){
        int md = (l + r) / 2;
        int idx = 0;
        graph g(n);
        while(idx < m and e[idx].second <= md){
            g[e[idx].first.first - 1].push_back({e[idx].first.second - 1, 0});
            g[e[idx].first.second - 1].push_back({e[idx].first.first - 1, 0});
            idx++;
        }
        REP(i, idx, m){
            g[e[i].first.first - 1].push_back({e[i].first.second - 1, 1});
            g[e[i].first.second - 1].push_back({e[i].first.first - 1, 1});
        }
        if((dijkstra(g, 0, n))[n - 1] <= k - 1){
            r = md;
        }else{
            l = md;
        }
    }
    cout << r << endl;
}
0