結果

問題 No.77 レンガのピラミッド
ユーザー yukirinyukirin
提出日時 2016-03-03 20:26:57
言語 Go
(1.23.4)
結果
WA  
実行時間 -
コード長 1,026 bytes
コンパイル時間 11,154 ms
コンパイル使用メモリ 240,556 KB
実行使用メモリ 6,824 KB
最終ジャッジ日時 2024-10-10 23:31:34
合計ジャッジ時間 12,001 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3 WA * 2
other AC * 2 WA * 18
権限があれば一括ダウンロードができます

ソースコード

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 = n/2 + 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]
	}

	count := excess - shortage
	count += shortage
	fmt.Println(count)
}

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

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