結果

問題 No.77 レンガのピラミッド
ユーザー tonyu0tonyu0
提出日時 2020-03-29 11:20:13
言語 Rust
(1.77.0 + proconio)
結果
AC  
実行時間 1 ms / 5,000 ms
コード長 969 bytes
コンパイル時間 11,247 ms
コンパイル使用メモリ 382,912 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-06-10 18:13:45
合計ジャッジ時間 12,181 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 1 ms
6,816 KB
testcase_01 AC 1 ms
6,812 KB
testcase_02 AC 0 ms
6,816 KB
testcase_03 AC 1 ms
6,940 KB
testcase_04 AC 0 ms
6,944 KB
testcase_05 AC 1 ms
6,944 KB
testcase_06 AC 1 ms
6,940 KB
testcase_07 AC 1 ms
6,940 KB
testcase_08 AC 1 ms
6,944 KB
testcase_09 AC 1 ms
6,940 KB
testcase_10 AC 1 ms
6,944 KB
testcase_11 AC 0 ms
6,940 KB
testcase_12 AC 0 ms
6,940 KB
testcase_13 AC 1 ms
6,940 KB
testcase_14 AC 1 ms
6,940 KB
testcase_15 AC 1 ms
6,944 KB
testcase_16 AC 0 ms
6,940 KB
testcase_17 AC 0 ms
6,940 KB
testcase_18 AC 1 ms
6,944 KB
testcase_19 AC 1 ms
6,940 KB
testcase_20 AC 1 ms
6,944 KB
testcase_21 AC 1 ms
6,944 KB
testcase_22 AC 1 ms
6,940 KB
testcase_23 AC 1 ms
6,944 KB
testcase_24 AC 1 ms
6,944 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

use std::io::*;

fn main() {
    let mut s: String = String::new();
    std::io::stdin().read_to_string(&mut s).ok();
    let mut itr = s.trim().split_whitespace();
    let n: usize = itr.next().unwrap().parse().unwrap();
    let mut a: Vec<usize> = (0..n)
        .map(|_| itr.next().unwrap().parse().unwrap())
        .collect();

    let sum = a.iter().fold(0, |acc, x| acc + x);
    let mut height = 1;
    while height * (height + 1) + height + 1 <= sum {
        height += 1;
    }

    let hidx = height - 1;
    a.resize(hidx * 2 + 1, 0);
    let excess = sum - height * (height - 1) - height;
    let mut need = 0;
    if a[hidx] < height {
        need = height - a[hidx];
    }
    for i in 0..hidx {
        if i < hidx * 2 + 1 && a[i] < i + 1 {
            need += i + 1 - a[i];
        }

        if hidx * 2 - i < hidx * 2 + 1 && a[hidx * 2 - i] < i + 1 {
            need += i + 1 - a[hidx * 2 - i];
        }
    }
    println!("{}", excess + need);
}
0