右上三角矩阵的压缩（Python实现）

目录

导语

解决方案

1. 以行为主的存储映射

2. 以列为主的存储映射 

示例题目

1. 题目描述

2. 输入/输出描述

3. 代码

4. 代码走读

传送门

测试用例

1、将如下的右上三角矩阵压缩

2、输入的行数数据测试

3、输入的默认值测试

4、输入的矩阵值测试

---

 

导语

上三角矩阵就是一种对角线以下元素都为0的 n x n矩阵。其中右上三角矩阵是指对角线以下的元素都为0。

由于上三角矩阵仍然有大量元素为0，为了避免浪费太多的内存空间，可以把三角矩阵的二维模式存储在一维列表中。

			...
0			...
0	0		...
...	...	...	...	...
0	0	0	...

对于如上所示的 n x n右上三角矩阵A(ij)，如果i > j，那么对应位置的元素值为0。

 

解决方案

由于采用Python解决这个问题，一维列表不需要实现声明大小。将该矩阵的有效元素映射为一维列表存储有两种解决方案，映射方式分为以行为主和以列为主两种方式。

1. 以行为主的存储映射

以行为主的存储方案是先将右上三角矩阵的第一行对角线及对角线以上的元素存储在一维列表中，然后在将右上三角矩阵的第二行对角线及对角线以上的元素存储在一维列表中。。。以此类推存储每一行元素。这样存储的一维列表中包含了右上三角矩阵中全部的有效元素。

...

...

...

 

观察一维列表存储的数据和坐标之间的关系，第0个位置存储，第1个位置存储，... 第n-1个位置存储。第n个位置存储...

因此存储在一维列表中的映射位置是

2. 以列为主的存储映射 

以列为主的存储方案是先将右上三角矩阵的第一列对角线及对角线以上的元素存储在一维列表中，然后在将右上三角矩阵的第二列对角线及对角线以上的元素存储在一维列表中。。。以此类推存储每一列元素。这样存储的一维列表中包含了右上三角矩阵中全部的有效元素。

存储原理与上文类似，这里直接给出映射关系：

---

无论选择哪一种存储映射，都可以将右上三角矩阵压缩为一个一维列表。本文选用以列为主的存储映射举例说明。

 

示例题目

1. 题目描述

设计一个Python程序，输入一个右上三角矩阵，输出压缩后的一维数组。

 

2. 输入/输出描述

输入描述：

输入右上三角矩阵的总行数、默认数字，然后自动计算大小，提示输入对应的数字。

Input matrix rows: 1
Input default value: 0
upper_triangular_matrix[0][0]: 8

输出描述：

输出原右上三角矩阵和压缩成一维列表后的矩阵。

------upper triangular matrix------
|	8	|
------compress upper triangular matrix --> list ------
[8]

 

3. 代码

class MatrixException(Exception):
    def __init__(self, message, code):
        self.message = message
        self.code = code


# 1. 输入一个右上三角矩阵
try:
    rows = int(input("Input matrix rows: "))
    if rows <= 0:
        raise MatrixException("Matrix rows can not less than zero.", 1002)
    columns = rows
    default_value = int(input("Input default value: "))

    utm = [[default_value] * columns for row in range(rows)]
    for row in range(rows):
        for column in range(columns):
            if column >= row:
                utm[row][column] = int(input("upper_triangular_matrix[%d][%d]: "
                                             % (row, column)))
except ValueError as e:
    print("errcode: %s" % str(1001))
    print("errmsg: %s" % str(e))
    exit(1001)
except MatrixException as e:
    print("errcode: %s" % e.code)
    print("errmsg: %s" % e.message)
    exit(e.code)


# 2. 压缩右上三角矩阵
try:
    array_size = ((1 + rows) * rows) // 2
    compress = [None] * array_size

    pos = 0
    for row in range(rows):
        for column in range(columns):
            if column >= row:
                compress[pos] = utm[row][column]
                pos += 1

    if None in compress:
        print("compress: %s" % compress)
        raise MatrixException("Compress upper triangular matrix failed.", 1002)
except MatrixException as e:
    print("errcode: %s" % e.code)
    print("errmsg: %s" % e.message)
    exit(e.code)

# 3. 打印结果
print("------upper triangular matrix------")

for row in range(rows):
    message = "|\t"
    for column in range(columns):
        message += str(utm[row][column]) + "\t"
    message += "|"
    print(message)

print("------compress upper triangular matrix --> list ------")
print(compress)

 

4. 代码走读

"""
Copyright TCatTime
"""

# 自定义压缩异常
class MatrixException(Exception):
    def __init__(self, message, code):
        self.message = message
        self.code = code


# 1. 输入一个右上三角矩阵
try:
    # 输入右上三角矩阵的行数，行数不大于0时报错
    rows = int(input("Input matrix rows: "))
    if rows <= 0:
        raise MatrixException("Matrix rows can not less than zero.", 1002)
    # 列数与行数保持相等
    columns = rows
    # 输入默认值（通常为0）
    default_value = int(input("Input default value: "))

    # 使用默认值初始化右上三角矩阵，并提示用户填充
    utm = [[default_value] * columns for row in range(rows)]
    for row in range(rows):
        for column in range(columns):
            if column >= row:
                utm[row][column] = int(input("upper_triangular_matrix[%d][%d]: "
                                             % (row, column)))
# 当用户输入的数据不是一个整数时，会报错并退出
except ValueError as e:
    print("errcode: %s" % str(1001))
    print("errmsg: %s" % str(e))
    exit(1001)
except MatrixException as e:
    print("errcode: %s" % e.code)
    print("errmsg: %s" % e.message)
    exit(e.code)


# 2. 压缩右上三角矩阵
try:
    # 初始化压缩列表，并使用None填充
    array_size = ((1 + rows) * rows) // 2
    compress = [None] * array_size

    # 将右上三角矩阵的数据写入压缩l
    pos = 0
    for row in range(rows):
        for column in range(columns):
            if column >= row:
                compress[pos] = utm[row][column]
                pos += 1

    if None in compress:
        print("compress: %s" % compress)
        raise MatrixException("Compress upper triangular matrix failed.", 1002)
except MatrixException as e:
    print("errcode: %s" % e.code)
    print("errmsg: %s" % e.message)
    exit(e.code)

# 3. 打印结果
print("------upper triangular matrix------")

for row in range(rows):
    message = "|\t"
    for column in range(columns):
        message += str(utm[row][column]) + "\t"
    message += "|"
    print(message)

print("------compress upper triangular matrix --> list ------")
print(compress)

 

传送门

input()函数

int()函数

range()函数

print()函数

exit()函数

ValueError

 

测试用例

1、将如下的右上三角矩阵压缩

5	6	7
0	3	9
0	0	1

Input matrix rows: 3
Input default value: 0
upper_triangular_matrix[0][0]: 5
upper_triangular_matrix[0][1]: 6
upper_triangular_matrix[0][2]: 7
upper_triangular_matrix[1][1]: 3
upper_triangular_matrix[1][2]: 9
upper_triangular_matrix[2][2]: 1
------upper triangular matrix------
|	5	6	7	|
|	0	3	9	|
|	0	0	1	|
------compress upper triangular matrix --> list ------
[5, 6, 7, 3, 9, 1]

 

2、输入的行数数据测试

1、输入0

Input matrix rows: 0
errcode: 1002
errmsg: Matrix rows can not less than zero.

2、输入负数

Input matrix rows: -4
errcode: 1002
errmsg: Matrix rows can not less than zero.

3、输入1

Input matrix rows: 1
Input default value: 0
upper_triangular_matrix[0][0]: 6
------upper triangular matrix------
|	6	|
------compress upper triangular matrix --> list ------
[6]

4、输入浮点数

Input matrix rows: 4.8
errcode: 1001
errmsg: invalid literal for int() with base 10: '4.8'

5、输入字符

Input matrix rows: s
errcode: 1001
errmsg: invalid literal for int() with base 10: 's'

 

3、输入的默认值测试

1、压缩如下矩阵。

5	6	7
13	3	9
13	13	1

 

将默认值default设置为13.

Input matrix rows: 3
Input default value: 13
upper_triangular_matrix[0][0]: 5
upper_triangular_matrix[0][1]: 6
upper_triangular_matrix[0][2]: 7
upper_triangular_matrix[1][1]: 3
upper_triangular_matrix[1][2]: 9
upper_triangular_matrix[2][2]: 1
------upper triangular matrix------
|	5	6	7	|
|	13	3	9	|
|	13	13	1	|
------compress upper triangular matrix --> list ------
[5, 6, 7, 3, 9, 1]

2、输入的默认值为浮点数

Input matrix rows: 3
Input default value: 3.6
errcode: 1001
errmsg: invalid literal for int() with base 10: '3.6'

3、输入的默认值为字符：

Input matrix rows: 3
Input default value: d
errcode: 1001
errmsg: invalid literal for int() with base 10: 'd'

 

4、输入的矩阵值测试

1、正确的测试值在上例已有演示。

2、矩阵值输入的是字符：

Input matrix rows: 3
Input default value: 0
upper_triangular_matrix[0][0]: a
errcode: 1001
errmsg: invalid literal for int() with base 10: 'a'

3、矩阵值输入的是浮点数：

Input matrix rows: 3
Input default value: 0
upper_triangular_matrix[0][0]: 1.3
errcode: 1001
errmsg: invalid literal for int() with base 10: '1.3'