Ссылка на оригинальную статью
Ссылка на Try Haxe!
Задача
Получить квадратную матрицу порядка [latex]n[/latex] [latex]\begin{pmatrix}1 &0 &\cdots & 0 \\ 0 & 1 &\cdots &0 \\ \cdots &\cdots &\cdots \cdots & \cdots \\ 0 & 0 & \cdots & 1\end{pmatrix}[/latex]
Тесты
n | Матрица |
---|---|
2 | [latex]\begin{pmatrix}1 & 0 \\ 0 & 1 \end{pmatrix}[/latex] |
4 | [latex]\begin{pmatrix}1 & 0 & 0 & 0 \\ 0 & 1 & 0 & 0 \\ 0 & 0 & 1 & 0 \\ 0 & 0 & 0 & 1 \end{pmatrix}[/latex] |
Код
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
class Main { static function main() { var n = Std.parseInt(Sys.stdin().readLine()); var a:Array<Array<Int>> = [for (x in 0...n) [for (y in 0...n) x == y ? 1 : 0]]; var out = Sys.stdout(); for (i in 0...n) { for (j in 0...n) { out.writeString(Std.string(a[i][j]) + " "); } out.writeString("\n"); } } } |
Решение
Из входного потока считываем $n$ — размер матрицы. Создаем матрицу со значением ноль по умолчанию. Затем в цикле заполняем главную диагональ единицами.