結果

問題 No.1211 円環はお断り
ユーザー risujiroh
提出日時 2020-08-30 14:12:23
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,263 bytes
コンパイル時間 4,814 ms
コンパイル使用メモリ 251,668 KB
最終ジャッジ日時 2025-01-13 21:40:29
ジャッジサーバーID
(参考情報)
judge1 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 31 TLE * 10
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/extc++.h>

#ifndef DUMP
#define DUMP(...) void(0)
#endif

using namespace std;

int main() {
  cin.tie(nullptr)->sync_with_stdio(false);
  int n, k;
  cin >> n >> k;
  vector<int64_t> pref(2 * n + 1);
  for (int i = 0; i < n; ++i) {
    int a;
    cin >> a;
    pref[i + 1] = pref[i] + a;
  }
  for (int i = n + 1; i <= 2 * n; ++i) pref[i] = pref[i - n] + pref[n];
  DUMP(pref);

  constexpr int lg = 17;
  auto check = [&](int64_t m) -> bool {
    vector to(lg, vector<int64_t>(n));
    for (int i = 0; i < n; ++i)
      to[0][i] = lower_bound(begin(pref), end(pref), pref[i] + m) - begin(pref);
    for (int s = 1; s < lg; ++s)
      for (int i = 0; i < n; ++i) {
        int ni = to[s - 1][i] % n;
        to[s][i] = to[s - 1][i] + (to[s - 1][ni] - ni);
      }
    for (int i = 0; i < n; ++i) {
      int t = k;
      int64_t cur = i;
      for (int s = lg; s--;) {
        if (t < 1 << s) continue;
        t -= 1 << s;
        cur += to[s][cur % n] - cur % n;
      }
      if (cur - i <= n) return true;
    }
    return false;
  };

  int64_t ok = 1, ng = pref.back() + 1;
  while (ng - ok > 1) {
    auto mid = (ok + ng) / 2;
    (check(mid) ? ok : ng) = mid;
  }
  cout << ok << '\n';
}
0