数据结构
网络
RDBMS
操作系统
Java
MS Excel
iOS
HTML
CSS
Android
Python
C 编程
C++
C#
MongoDB
MySQL
Javascript
PHP
物理
化学
生物
数学
英语
经济
心理学
社会科学
服装科学
法律科学Python 程序查找二维数组中的 k'th 最小元素
一个 n×n 的用户输入整数矩阵已给出和 k 的值。我们的任务是找出二维数组中的 k'th 最小元素。这里我们使用 heapq 模块。Python 中的堆队列（或 heapq）。在 Python 中，它可用“heapq”模块。该模块在 Python 中的技术是每次弹出堆元素中最小的（最小堆）。nsmallest() 方法用于从数据帧或序列中获取 n 个最小值。
示例
Input Array is:: 10 20 20 40 15 45 40 30 32 33 30 50 12 78 99 78 The value of k is 10 10 th smallest element is 40
算法
Step 1: First create a 2D array. Step 2: Then assign first row to a variable and convert it into min heap. Step 3: Then traverse remaining rows and push elements in min heap. Step 4: Now use nsmallest(k, iterable) method of heapq module and get list of first k smallest element, nsmallest(k,list) method returns first k smallest element now print last element of that list.
示例代码
# python program to find K'th smallest element in
# a 2D array in Python
import heapq
def smallestele(A):
assignval = A[0]
heapq.heapify(assignval)
for i in A[1:]:
for j in i:
heapq.heappush(assignval,j)
mini = heapq.nsmallest(k,assignval)
print (k,"th smallest element is ",mini[-1])
# Driver program
if __name__ == "__main__":
A=[]
n=int(input("Enter N for N x N matrix : ")) #3 here
#use list for storing 2D array
#get the user input and store it in list (here IN : 1 to 9)
print("Enter the element ::>")
for i in range(n):
row=[] #temporary list to store the row
for j in range(n):
row.append(int(input())) #add the input to row list
A.append(row) #add the row to the list
print(A)
# [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
#Display the 2D array
print("Display Array In Matrix Form")
for i in range(n):
for j in range(n):
print(A[i][j], end=" ")
print() #new line
k = int(input("Enter the kth position ::>"))
smallestele(A)
输出
Enter N for N x N matrix : 4 Enter the element ::> 10 20 20 40 15 45 40 30 32 33 30 50 12 78 99 78 [[10, 20, 20, 40], [15, 45, 40, 30], [32, 33, 30, 50], [12, 78, 99, 78]] Display Array In Matrix Form 10 20 20 40 15 45 40 30 32 33 30 50 12 78 99 78 Enter the kth position ::>10 10 th smallest element is 40
更新于： 30-Jul-2019
400 浏览量
广告