New Go

BasicSyntax/prime/transame.go

@@ -0,0 +1,31 @@
+package main
+
+import "fmt"
+
+func transpose(matrix [][]int) bool {
+	for i := range matrix {
+		for j := range matrix[i] {
+			if matrix[i][j] != matrix[j][i] {
+				return false
+			}
+		}
+	}
+	return true
+}
+
+func main() {
+	var n int
+	fmt.Scan(&n)
+	matrix := make([][]int, n)
+	for i := range matrix {
+		matrix[i] = make([]int, n)
+		for j := range matrix[i] {
+			fmt.Scan(&matrix[i][j])
+		}
+	}
+	if transpose(matrix) {
+		fmt.Println("YES")
+	} else {
+		fmt.Println("NO")
+	}
+}