結果

問題 No.78 クジ付きアイスバー
コンテスト
ユーザー Mister
提出日時 2020-04-22 10:02:09
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,261 bytes
コンパイル時間 3,517 ms
コンパイル使用メモリ 75,136 KB
最終ジャッジ日時 2025-01-09 22:24:54
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 35
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <string>

using lint = long long;
constexpr lint INF = 1LL << 40;

template <class T>
std::vector<T> vec(int len, T elem) { return std::vector<T>(len, elem); }

void solve() {
    int n;
    lint k;
    std::string s;
    std::cin >> n >> k >> s;

    auto dp = vec(40, vec(n, 0LL));
    for (int i = 0; i < n; ++i) {
        lint get = 0, cnt = 1;
        for (int j = 0; j < n; ++j) {
            ++get;
            cnt += int(s[(i + j) % n] - '0') - 1;
            if (cnt == 0) break;
        }
        dp[0][i] = (cnt > 0 ? INF : get);
    }

    for (int j = 1; j < 40; ++j) {
        for (int i = 0; i < n; ++i) {
            if (dp[j - 1][i] == INF) {
                dp[j][i] = INF;
            } else {
                int ni = (i + dp[j - 1][i]) % n;
                dp[j][i] = std::min(dp[j - 1][i] + dp[j - 1][ni], INF);
            }
        }
    }

    lint ans = 0;
    int i = 0;
    for (int j = 39; j >= 0; --j) {
        if (dp[j][i] > k) continue;
        ans += (1LL << j);
        k -= dp[j][i];
        (i += dp[j][i]) %= n;
    }

    std::cout << ans + (k > 0) << std::endl;
}

int main() {
    std::cin.tie(nullptr);
    std::ios::sync_with_stdio(false);

    solve();

    return 0;
}
0