CodeIQ：進捗ビッグバンマーク

私自身が表題の問題を解いた時のプログラムについて解説します。
問題の詳細は「進捗ビッグバンマーク」（CodeIQ）を参照してください。

問題の概要

問題を引用します。
進捗絶大にあったことを知らせる「進捗ビッグバンマーク」を作ることにしました。
「進捗ビッグバンマーク」は高さ X 幅 X の正方形の中に B のカタチをした文字を半角小文字の「b」で描きます。
1行目、真ん中の行、最終行はすべての列が「b」です。
それ以外の行の1列目と最終列は「b」です。
それ以外を半角ドット「.」にします。

例 高さ 1 幅 1 の進捗ないですマーク

b

例 高さ 3 幅 3 の進捗ないですマーク

bbb
bbb
bbb

例 高さ 5 幅 5 の進捗プログレスマーク

bbbbb
b...b
bbbbb
b...b
bbbbb

「進捗ビッグバンマーク」を手入力するのはしんどいのでプログラムで出力することにしました。
「進捗ビッグバンマーク」を出力するプログラムを実装してください。

仕様
・標準入力の高さ（＝幅）を元に「進捗ビッグバンマーク」を出力する

標準入力
・標準入力の高さが 1-100 の整数で与えられます

例:
5

標準出力
・高さが奇数の場合は「進捗ないですマーク」を出力する
・高さが偶数の場合は「invalid」を出力する

例（入力の例に対する出力の例）

bbbbb
b...b
bbbbb
b...b
bbbbb

その他の仕様
・標準入力の末尾には改行があります
・標準出力の末尾に改行をつけてください
・標準入力の仕様で説明した内容以外の入力は行われません（不正入力に対するチェックは不要）

Samples
Sample1
Input
1

b

Sample2
Input
3

bbb
bbb
bbb

Sample3
Input
5

bbbbb
b...b
bbbbb
b...b
bbbbb
Sample4
Input
2

invalid

私のプログラム

Rubyで解答しています。

#!/usr/bin/ruby

def solve(n)
	if n%2 == 0 then
		puts "invalid"
		return
	end

	for i in 0...n
		if (i == 0) || (i == n/2) || (i == n-1) then l = 'b' * n
		else l = 'b' + '.'*(n-2) + 'b'
		end
		puts l
	end
	return
end

# main
while line = gets
	line.strip!
	next if line.empty?
	solve(line.to_i)
end

解説

「進捗マーク」シリーズです

main

入力値を数値にしてsolve()に渡します。

solve(n)

i==n, i==n/2, i==n-1の場合はn個のb、それ以外は両端をbとその間にn-2個の.にします。

雑感

特にありません。