Problem:
イルミネーション (Illumination)

問題

JOI 社の建物は図のような 1 辺 1 メートルの正六角形をつなぎ合わせた形である．クリスマスが近づいているので，JOI 社では建物の壁面をイルミネーションで飾り付けることにした．ただし，外から見えない部分にイルミネーションを施すのは無駄なので，イルミネーションは外から建物の中を通らずに行くことのできる壁面にのみ飾り付けることにした．

図: JOI 社の建物の配置の例

上の図は上空から見た JOI 社の建物の配置の例である．正六角形内の数字は座標を表す．灰色の正六角形は建物がある場所を表し，白色の正六角形は建物がない場所を表す．この例では，赤の実線で示される部分がイルミネーションで飾り付けを行う壁面となり，その壁面の長さの合計は 64 メートルとなる．

JOI 社の建物の配置を表す地図が与えられたとき，飾り付けを行う壁面の長さの合計を求めるプログラムを作成せよ．ただし，地図の外側は自由に行き来できるものとし，隣接した建物の間は通ることはできないものとする．

入力

入力ファイルの 1 行目には 2 つの整数 W, H (1 ≦ W ≦ 100，1 ≦ H ≦ 100) が空白を区切りとして書かれている．続く H 行には JOI 社の建物の配置が書かれている．i + 1 行目 (1 ≦ i ≦ H) には W 個の整数が空白を区切りとして書かれており，j 個目 (1 ≦ j ≦ W) の整数は座標 (j, i) の正六角形に建物がある時は 1 であり，ない時は 0 である．また，与えられる入力データには建物が必ず 1 つ以上ある．

地図は以下の規則によって記述されている．

地図の最も北の行の最も西の正六角形は座標 (1, 1) である．

座標 (x, y) の正六角形に隣接する東隣の正六角形は座標 (x + 1, y) である．

y が奇数の時，座標 (x, y) の正六角形に隣接する南西の正六角形の座標は (x, y + 1) である．

y が偶数の時，座標 (x, y) の正六角形に隣接する南東の正六角形の座標は (x, y + 1) である．

出力

イルミネーションで飾り付けを行う壁面の長さの合計を 1 行で出力せよ．

入出力例

入力例 1

8 4
0 1 0 1 0 1 1 1
0 1 1 0 0 1 0 0
1 0 1 0 1 1 1 1
0 1 1 0 1 0 1 0

出力例 1

64

入出力例 1 は問題文中の例に対応しており，イルミネーションで飾り付けを行う壁面の長さの合計は 64 メートルである．

入力例 2

8 5
0 1 1 1 0 1 1 1
0 1 0 0 1 1 0 0
1 0 0 1 1 1 1 1
0 1 0 1 1 0 1 0
0 1 1 0 1 1 0 0

出力例 2

56

問題文と自動審判に使われるデータは、情報オリンピック日本委員会が作成し公開している問題文と採点用テストデータです。

Solution in Kotlin (judge verdict: Accepted)

import java.util.*

fun main(args: Array<String>?): Unit {
    val (width, height) = readLine()!!.trim().split(' ').map(String::toInt)
    val state = Array(height){ readLine()!!.trim().split(' ').map(String::toInt).toIntArray()}
    val queue = LinkedList<HexPoint>()
    for (h in 0 until height) {
        if (state[h][0] == 0) queue.push(HexPoint(0, h))
        if (state[h].last() == 0) queue.push(HexPoint(state[h].lastIndex, h))
    }
    for (w in 0 until width){
        if (state[0][w] == 0) queue.push(HexPoint(w, 0))
        if (state.last()[w] == 0) queue.push(HexPoint(w, state.lastIndex))
    }
    for (hp in queue) state[hp.y][hp.x] = 2
    while(queue.isNotEmpty()) {
        val current = queue.pop()
        for (next in current.neighbors()) if (next.x in 0 until width && next.y in 0 until height && state[next.y][next.x] == 0){
            state[next.y][next.x] = 2
            queue.push(next)
        }
    }
    println(
            (0 until width).sumBy{ x -> (0 until height).filter{y -> state[y][x] == 1}.sumBy { y -> HexPoint(x, y).neighbors().count{ hp -> hp.x !in 0 until width || hp.y !in 0 until height || state[hp.y][hp.x] == 2 } }}
    )

}
data class HexPoint(val x: Int, val y: Int) {
    fun neighbors(): List<HexPoint> = when(y % 2 == 0){
        true -> listOf(HexPoint(x, y - 1), HexPoint(x + 1, y - 1), HexPoint(x - 1, y), HexPoint(x + 1, y), HexPoint(x, y + 1), HexPoint(x + 1, y + 1))
        false -> listOf(HexPoint(x - 1, y - 1), HexPoint(x, y - 1), HexPoint(x - 1, y), HexPoint(x + 1, y), HexPoint(x - 1, y + 1), HexPoint(x, y + 1))
    }
}