結果

問題 No.77 レンガのピラミッド
コンテスト
ユーザー DialBird
提出日時 2017-02-22 09:36:30
言語 C++11（廃止可能性あり）
(gcc 13.3.0)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,228 bytes
コンパイル時間 1,734 ms
コンパイル使用メモリ 159,984 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-12-30 18:37:38
合計ジャッジ時間 3,019 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 5
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

using namespace std;

#define REP(i,first,last) for (int i=first;i<last;++i)
#define MAX(x,y) (x > y ? x : y)
#define MIN(x,y) (x < y ? x : y)

int N;

int main(){
  cin >> N;
  vector<int> blocks(N);
  int sum = 0;

  REP(i,0,N){
    cin >> blocks[i];
    sum += blocks[i];
  }

  // ピラミッド構成ブロックの数はnの二乗になり、それぞれのピラミッドの最も高い列はn個のブロックになる
  // そして、与えられたブロックの合計値でできるもっとも大きいピラミッドを作るのが最短
  // この作業は各列の「余っているブロック」のみ動かすので、余っているブロックをカウントすればいい。
  int p_top_num = sqrt(sum);
  int p_row_cnt = 1 + (p_top_num - 1) * 2;

  int required = 1;
  int amari = 0;
  REP(i,0,p_top_num){
    if (i < N) {
      if (blocks[i] >= required) {
        amari += blocks[i] - required;
      }
    }
    ++required;
  }

  required = p_top_num - 1;
  REP(i,p_top_num,p_row_cnt){
    if (i < N && required <= blocks[i]) {
        amari += blocks[i] - required;
    }
    --required;
  }

  REP(i,p_row_cnt,N){
    amari += blocks[i];
  }

  cout << amari <<endl;
}
0