无序数组中最小的k个数

题目描述

对于一个无序数组，数组中元素为互不相同的整数，请返回其中最小的k个数，顺序与原数组中元素顺序一致。

给定一个整数数组A及它的大小n，同时给定k，请返回其中最小的k个数。

测试样例：

[1,2,4,3],4,2

返回：[1,2]

简单题

# -*- coding:utf-8 -*-

class KthNumbers:
    def findKthNumbers(self, A, n, k):
        # write code here
        newA = sorted(A)
        st = newA[:k]
        li = []
        for x in A:
        	if x in st:
        		li.append(x)

        return li

if __name__ == "__main__":
	a = KthNumbers()
	print a.findKthNumbers([2,1,4,3],4,2)