OJ#189.实现折半查找

题目描述

​ 我们刚刚学习了折半查找，但是还没有自己来实践一下，这一节，你就需要自己实现一个折半查找的程序啦。​ 给定 N个整数和 K个待查找的整数 M1,M2,…,Mk。
如果待查找的整数在给定的 N个整数中，请输出待查找的整数是数组中第几个元素（从 1 开始计算，第一个元素计 1 而不是0 ）；
​如果待查找的整数不在给定的 N 个整数中，则输出 0 。

输入

​ 测评机可能会反复运行程序，每次程序运行后，测评机输入包括三行。

​ 第一行输入两个整数N（ 1≤N≤106）和 K（1≤K≤106），分别表示给定的整数总个数和待查找的数的个数；

​ 第二行从小到大输入 N 个整数 Numbersi（ 1≤Numbersi≤106，其中 1≤i≤N），每两个整数之间用一个空格隔开；

​ 第三行输入 K个整数 Numbersj （ 1≤Numbersj≤106，其中 1≤j≤K），每两个整数之间用一个空格隔开。

输出

​ 输出为一行，包括 K个部分，每个部分为题目描述中待查找的元素索引或 0 ，K个部分之间用一个空格分隔。

​ 输出行尾不含多余空格。

样例输入1

3 1
1 4 6
4

样例输出1

2

样例输入2

5 2
1 4 6 7 8
5 1

样例输出2

0 1

代码

// 静态数组和动态申请内存的方式都可以，执行时间差不多
#include <stdio.h>
#include <stdlib.h>
// #define N 1000005

// int arrOrigin[N] = {0};
// int arrFound[N] = {0};

int binarySearch (int *arr, int n, int x) {
  int mid, head = 0, tail = n - 1;

  while (head <= tail) {
    mid = head + tail >> 1;
    if (x == arr[mid]) return mid;
    if (x < arr[mid]) tail = mid - 1;
    else head = mid + 1;
  }
  return -1;
}

int main () {
  int n, k, *arrOrigin, *arrFound;
  // int arrOrigin[N] = 0, arrFound[N] = 0;
  scanf("%d%d", &n, &k);
  arrOrigin = (int *)calloc(n, sizeof(int));
  arrFound = (int *)calloc(k, sizeof(int));
  for (int i = 0; i < n; i++) {
    scanf("%d", &arrOrigin[i]);
  }

  for (int j = 0; j < k; j++) {
    scanf("%d", &arrFound[j]);
  }
  
  for (int i = 0; i < k; i++) {
    int pos = binarySearch(arrOrigin, n ,arrFound[i]);
    if(pos < 0) {
      i < k - 1 ? printf("%d ", 0) : printf("%d", 0);
    } else {
      i < k - 1 ? printf("%d ", pos + 1) : printf("%d", pos + 1);
    } 
  }
  free(arrOrigin);
  free(arrFound);
  return 0;
}