โปรแกรมหาตัวเลขน้อยสุดอันดับ n จากเมทริกซ์ใน Python

สมมติว่ามีเมทริกซ์ 2D ที่แต่ละแถวและคอลัมน์เรียงจากน้อยไปมาก ต้องการหาตัวเลขน้อยสุดอันดับ n

ตัวอย่าง: เมทริกซ์เดียวกัน n = 4 ผลลัพธ์คือ 6

ขั้นตอนการแก้ปัญหา

1. รวมทุกค่าในเมทริกซ์เข้า list lst
2. เรียง lst
3. คืน lst[n]

ตัวอย่างโค้ด

class Solution:
    def solve(self, matrix, n):
        lst = []
        for i in matrix:
            for j in i:
                lst.append(j)
        lst.sort()
        return lst[n]

ob = Solution()
matrix = [[2, 4, 30], [3, 4, 31], [6, 6, 32]]
n = 4
print(ob.solve(matrix, n))

อินพุต

matrix = [[2, 4, 30], [3, 4, 31], [6, 6, 32]] n = 4

เอาต์พุต

6