首页 > 试题广场 >

KiKi判断上三角矩阵

[编程题]KiKi判断上三角矩阵
  • 热度指数:16401 时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 256M,其他语言512M
  • 算法知识视频讲解

KiKi想知道一个n阶方矩是否为上三角矩阵,请帮他编程判定。上三角矩阵即主对角线以下的元素都为0的矩阵,主对角线为从矩阵的左上角至右下角的连线。


输入描述:

第一行包含一个整数n,表示一个方阵包含n行n列,用空格分隔。 (2≤n≤10)

从2到n+1行,每行输入n个整数(范围-231~231-1),用空格分隔,共输入n*n个数。



输出描述:

一行,如果输入方阵是上三角矩阵输出"YES"并换行,否则输出"NO"并换行。

示例1

输入

3
1 2 3
0 4 5
0 0 6

输出

YES
示例2

输入

4
1 2 3 4
5 6 7 8
9 0 11 12 
13 0 0 16

输出

NO
在参考第一名的回答之后作出了一些简单的优化:
题目的n已经给定范围了,2<n
所以可以省去n=1时候的情形;
n = int(input())
array = []
for i in range(n):
    array.append([int(x) for x in input().split()])
flag =1
for x in range(1,n):#行
    for y in range(x) :#列
        if (array[x][y] != 0):
            flag =0
if flag ==1:
    print("YES")
else:
    print("NO")

发表于 2020-10-12 11:17:03 回复(0)