結果

問題 No.929 よくあるボールを移動するやつ
コンテスト
ユーザー sash2104
提出日時 2019-11-22 22:06:58
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 749 bytes
コンパイル時間 583 ms
コンパイル使用メモリ 69,260 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-10-11 03:46:20
合計ジャッジ時間 1,330 ms
ジャッジサーバーID
（参考情報） 		judge5 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 9 WA * 3
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <cassert>
#include <iostream>
#include <vector>
using namespace std;

int main() {
  int n;
  cin >> n;
  vector<int> lb(n);
  for (int &b: lb) cin >> b;

  int ans = 0;
  int cnt = 0;
  for (int i = 0; i < n-1; ++i) {
    cnt += lb[i];
    // cout << i << " " << lb[i] << " " << cnt << endl;
    if (cnt > i+1) {
      assert(lb[i] > 1);
      int diff = cnt-(i+1);
      lb[i+1] += diff;
      lb[i] -= diff;
      ans += diff;
      cnt -= diff;
    }
  }
  cnt = 0;
  for (int i = n-1; i > 0; --i) {
    cnt += lb[i];
    if (cnt > n-i) {
      assert(lb[i] > 1);
      int diff = cnt-(n-i);
      lb[i-1] += diff;
      lb[i] -= diff;
      assert(lb[i] == 1);
      ans += diff;
      cnt -= diff;
    }
  }
  cout << ans << endl;
}
0