結果

問題 No.2712 Play more!
コンテスト
ユーザー blue_jam
提出日時 2024-03-31 15:20:39
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 3,203 bytes
コンパイル時間 4,698 ms
コンパイル使用メモリ 257,012 KB
最終ジャッジ日時 2025-02-20 17:51:34
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 32 WA * 1
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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

using namespace std;
using namespace atcoder;

using ll = long long;
using mint = modint998244353;

ll solveF(ll H, ll W, ll K) {
    vector<vector<mint>> dp(W + 1, vector<mint>(1 << H, 0));
    dp[0][(1 << H) - 1] = 1LL;
    for (ll j = 0; j < W; j++) {
        for (ll s = 0; s < (1 << H); s++) {
            for (ll t = 0; t < (1 << H); t++) {
                ll u = s & t;
                ll cnt = __builtin_popcount(u);
                if (cnt >= K) {
                    dp[j + 1][t] += dp[j][s];
                }
            }
        }
    }
    mint ans = 0;
    for (ll s = 0; s < (1 << H); s++) {
        ans += dp[W][s];
    }
    return ans.val();
}

ll solveFNaive(ll H, ll W, ll K) {
    return 0;
}

constexpr long long INF = (1LL << 60);

// 辺
struct Edge
{
    int from;
    int to;
    ll cost;
};

// ベルマンフォード法 (1.2 負閉路の影響を受ける頂点を調べる)
// 負の閉路が存在する場合 true を返し, 負閉路の影響を受ける頂点は -INF にセットされる
// distances は頂点数と同じサイズ, 全要素 INF で初期化しておく
bool BellmanFord(const std::vector<Edge>& edges, std::vector<long long>& distances, int startIndex)
{
    distances[startIndex] = 0;

    for (size_t i = 0; i < distances.size(); ++i)
    {
        bool changed = false;

        // 各辺について
        for (const auto& edge : edges)
        {
            // (INF + cost) は INF なので処理しない
            if (distances[edge.from] == INF)
            {
                continue;
            }

            // to までの新しい距離
            const long long d = (distances[edge.from] + edge.cost);

            // d が現在の記録より小さければ更新
            if (d < distances[edge.to])
            {
                distances[edge.to] = d;

                changed = true;
            }
        }

        // どの頂点も更新されなかったら終了
        if (!changed)
        {
            return false;
        }
    }

    // 頂点数分だけさらに繰り返し, 負閉路の影響を受ける頂点に -INF を伝播
    for (size_t i = 0; i < distances.size(); ++i)
    {
        for (const auto& edge : edges)
        {
            if (distances[edge.from] == INF)
            {
                continue;
            }

            const long long d = (distances[edge.from] + edge.cost);

            if (d < distances[edge.to])
            {
                // 負閉路の影響を受ける頂点を -INF に
                distances[edge.to] = -INF;
            }
        }
    }

    return true;
}

int main() {
    ll N, M;
    cin >> N >> M;
    vector<ll> A(N);
    for (ll i = 0; i < N; i++) {
        cin >> A[i];
    }

    vector<Edge> edges(M);
    for (ll i = 0; i < M; i++) {
        ll a, b, c;
        cin >> a >> b >> c;
        a--; b--;
        edges[i] = {(int)a, (int)b, c - A[b]};
    }

    vector<ll> dist(N);
    bool negativeRoop = BellmanFord(edges, dist, 0);

    if (dist[N - 1] == -INF) {
        cout << "inf" << endl;
    } else {
        cout << A[0] - dist[N - 1] << endl;
    }

    return 0;
}
0