【剑指Offer面试编程题】题目1371：最小的K个数--九度OJ

题目描述：: 输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

输入：

每个测试案例包括2行：

第一行为2个整数n，k(1<=n，k<=200000)，表示数组的长度。

第二行包含n个整数，表示这n个数，数组中的数的范围是[0,1000 000 000]。

输出：: 对应每个测试案例，输出最小的k个数，并按从小到大顺序打印。

样例输入：

8 4

4 5 1 6 2 7 3 8

样例输出：

1 2 3 4

【解题思路】本题首先想到的是利用二分查找确定一个元素的位置，然后比较这个位置是在k之前还是之后，若这个位置等于k则停止，否则继续在分开的两个子区间里面确定k位置的元素，直到确定k位置元素后停止。然后对前k个元素排序即可得到前k个有序序列。

但考虑到上面的做法代码比较复杂，且要求通用的话，时间复杂度也挺高。所以决定用一个笨办法试试，直接对所有元素排序，然后取前k个元素输出即可。

AC code：

#include <cstdio>

#include <vector>

#include <algorithm>

using namespace std;

int main()

{

  int n,k;

  while(scanf("%d%d",&n,&k)!=EOF)

  {

    vector<int> vecin(n);

    int tt;

    for(int i=0;i<n;++i)

      scanf("%d",&vecin[i]);

    sort(vecin.begin(),vecin.end());

    printf("%d",vecin[0]);

    for(int i=1;i<k && i<n;++i)

      printf(" %d",vecin[i]);

    printf("\n");

  }

  return 0;

}

/**************************************************************

    Problem: 1371

    User: huo_yao

    Language: C++

    Result: Accepted

    Time:850 ms

    Memory:1836 kb

****************************************************************/

题目链接：http://ac.jobdu.com/problem.php?pid=1371

九度-剑指Offer习题全套答案下载：http://download.csdn.net/detail/huoyaotl123/8276299

巴特西

【剑指Offer面试编程题】题目1371：最小的K个数--九度OJ

最新文章

热门文章