返回矩阵每行最后K个非零元素的索引

# 定义一个函数，接受矩阵和K作为输入参数
def find_last_k_nonzero_indices(matrix, K):
    result = []  # 存储结果的列表
    for row in matrix:
        nonzero_indices = []  # 存储非零元素的索引的列表
        for i in range(len(row)-1, -1, -1):  # 从最后一个元素开始遍历
            if row[i] != 0:  # 如果当前元素不为零
                nonzero_indices.append(i)  # 将索引添加到列表中
                if len(nonzero_indices) == K:  # 如果找到了K个非零元素的索引
                    break  # 结束内层循环
        result.append(nonzero_indices)  # 将当前行的非零元素索引列表添加到结果中
    return result

# 调用函数并打印结果
matrix = [[1, 2, 0, 3, 0], [0, 4, 5, 0, 6], [7, 0, 8, 9, 0]]
K = 2
result = find_last_k_nonzero_indices(matrix, K)
print(result)