【剑指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("
");
  }
  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