CodeIQ：進捗やったったマーク

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

問題の概要

問題を引用します。

進捗進んだ！やった！！というきもちを知らせる「進捗やったったマーク」を作ることにしました。
「進捗やったったマーク」は高さ X 幅 X の正方形の中にYのカタチをした文字を半角小文字の「y」で描きます。
真ん中の行より上は左右の(行数)の位置が「y」です。
例えば1行目なら左から1列目、右から1列目が「y」、2行目なら左から2列目、右から2列目が「y」です。
それ以外を半角ドット「.」にします。

例高さ 1 幅 1 の進捗はひみつマーク

y

例高さ 3 幅 3 の進捗はひみつマーク

y.y
.y.
.y.

例高さ 5 幅 5 の進捗はひみつマーク

y...y
.y.y.
..y..
..y..
..y..

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

仕様
・標準入力の高さ（＝幅）を元に「進捗やったったマーク」を出力する

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

例:

5

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

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

y...y
.y.y.
..y..
..y..
..y..

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

Samples
Sample1
Input

1

y

Sample2
Input

3

y.y
.y.
.y.

Sample3
Input

5

y...y
.y.y.
..y..
..y..
..y..

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
		str = '.' * n

		if i < n/2 then
			str[i] = 'y'
			str[n-1-i] = 'y'
		else
			str[n/2] = 'y'
		end
		puts str
	end
end

# main
while line = gets
	line.strip!
	next if line.empty?

	solve(line.to_i)
end

解説

「進捗マーク」シリーズ、今回はyでした。

main

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

solve(n)

「進捗ダメです」と基本的に同じです。
n/2行目までは両端からi番目をyに、それ以降はn/2番目をyにします。

雑感

まだAはやりそうな気がします。Nもあるかな？