rhombus

rhombus.go

package main

import "fmt"

func main() {
	n := 0
	fmt.Scanf("%d\n", &n)

	arr := make([][]int, n)
	for i := 0; i < n; i++ {
		arr[i] = make([]int, n)
	}

	cur := 1
	x, y := 0, 0
	for {
		if n*n < cur {
			break
		}
		for p, q := x, y; ; p, q = p-1, q+1 {
			arr[p][q] = cur
			cur++
			if p == 0 {
				if q == n-1 {
					y++
					break
				}
				x++
				break
			}
			if q == n-1 {
				y++
				break
			}
		}
	}

	for i := 0; i < n; i++ {
		for j := 0; j < n; j++ {
			fmt.Printf("%3d ", arr[j][i])
		}
		fmt.Println()
	}
}