牛客網 劍指offer 最小的k個數

輸入n個整數,找出其中最小的K個數。例如輸入4,5,1,6,2,7,3,8這8個數字,則最小的4個數字是1,2,3,4,。

上代碼

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        int len = input.size();
        vector<int> arr;
        for(int i = len-1;i>0;i--)
        {
            bool flag = true;
            for(int j=0;j<i;j++)
            {
                if (input[j+1]<input[j])
                {
                    int temp =input[j+1];
                    input[j+1]=input[j];
                    input[j]=temp;
                    flag = false;
                }
            }
            if (flag) break;
        }
        if(k>len||k<=0) return arr;
        for(int i=0;i<k;i++)
        {
           arr.push_back(input[i]);
        }
        return arr;
    }
};
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章