結果

問題 No.77 レンガのピラミッド
コンテスト
ユーザー yukirin
提出日時 2016-03-03 20:43:30
言語 Go
(1.23.4)
結果
WA  
実行時間 -
コード長 985 bytes
コンパイル時間 11,603 ms
コンパイル使用メモリ 232,732 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-10-10 23:32:32
合計ジャッジ時間 12,463 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 1 WA * 4
other AC * 1 WA * 19
権限があれば一括ダウンロードができます

ソースコード

diff # 

package main

import (
	"bufio"
	"fmt"
	"os"
	"strconv"
)

var sc = bufio.NewScanner(os.Stdin)
var rdr = bufio.NewReaderSize(os.Stdin, 1000000)

func main() {
	sc.Split(bufio.ScanWords)
	n := nextInt()
	ns, sum := make([]int, n), 0
	for i := range ns {
		c := nextInt()
		sum += c
		ns[i] = c
	}

	h := 0
	for i := 1; ; i++ {
		if i*i > sum {
			h = i - 1
			if h*2-1 > len(ns) {
				h = len(ns)/2 + (n & 1)
			}
			break
		}
	}

	excess, shortage := 0, 0
	if h > ns[h-1] {
		shortage += h - ns[h-1]
	} else {
		excess += ns[h-1] - h
	}

	for i := 0; i < h-1; i++ {
		if i+1 > ns[i] {
			shortage += i + 1 - ns[i]
		} else {
			excess += ns[i] - (i + 1)
		}

		if h-i-1 > ns[i+h] {
			shortage += (h - i - 1) - ns[i+h]
		} else {
			excess += ns[i+h] - (h - i - 1)
		}
	}

	for i := h*2 - 1; i < len(ns); i++ {
		excess += ns[i]
	}

	fmt.Println(excess / 2)
}

func nextLine() string {
	sc.Scan()
	return sc.Text()
}

func nextInt() int {
	i, _ := strconv.Atoi(nextLine())
	return i
}
0