查找最小的k个元素

题目:输入n个整数,输出其中最小的k个。
例如输入1,2,3,4,5,6,7和8这8个数字,则最小的4个数字为1,2,3和4。

a = [9,8,5,2,7,1,10,126,100]

def test a,k
  if a.length<=k
    p a
    return
  end
  b = []
  max = a[0]
  a.each{|v|
    if b.length<k 
      b<<v
      max = v if v>max
    else
      if v<max
        b.delete_at(b.index(b.max))
        b<<v
        max = b.max
      end
    end
  }
  p b
end

test a,7

 

posted on 2013-09-16 17:12  张飞_  阅读(282)  评论(0编辑  收藏  举报

导航