查找问题归纳

例题1 找x （哈尔滨工业大学复试上机题）

题目描述：

输入一个数n，然后输入n个数值各不相同，再输入一个值x，输出这个值在这个数组中的下标（从0开始，若不在数组中则输出-1）。

输入描述：

测试数据有多组，输入n(1<=n<=200)，接着输入n个数，然后输入x。

输出描述：

对于每组输入,请输出结果。

输入：

2

1 3

0

输出：

-1

代码示例：

//顺序查找
#include <cstdio>
int main(){
    int n;
    int arr[200];
    scanf("%d", &n);
    for (int i = 0; i < n; ++i) {
        scanf("%d", &arr[i]);
    }
    int x;
    scanf("%d", &x);
    //查找的操作
    int idx;
    for (idx = 0; idx < n; ++idx) {
        if(arr[idx] == x){
            printf("%d\n", idx);
            break;
        }
    }
    if(idx == n){
        printf("-1\n");
    }
}

分析：

本题是最简单的顺序查找问题，若数组中有n个元素，要查m个数，那么所需时间复杂度要O(m*n)，可见顺序查找数据多时效率不高。

例题2 查找 （北京邮电大学复试上机题）

题目描述：

输入数组长度 n 输入数组 a[1...n] 输入查找个数m 输入查找数字b[1...m] 输出 YES or NO 查找有则YES 否则NO 。

输入描述：

输入有多组数据。 每组输入n，然后输入n个整数，再输入m，然后再输入m个整数（1<=m,n<=100）。

输出描述：

如果在n个数组中输出YES否则输出NO。

输入：

5

1 5 2 4 3

3

2 5 6

输出：

YES

YES

NO

代码示例1：

//二分查找
#include <cstdio>
#include <algorithm>
using namespace std;
int arr[100];//全局的数组，方便在不同的函数间共享

bool binarySearch(int n, int x){
    //查到了返回true，查不到返回false
    int left = 0;
    int right = n-1;
    while (left <= right){ //出循环条件！！！
        int mid = (left + right)/2;
        if(arr[mid] == x){
            return true;
        }
        else if(arr[mid] > x){
            right = mid - 1;//右边缘往左边缩
            //最后的边界情况 right和left相等，下次right可能会变成left-1
        }
        else{
            left = mid + 1;//左边缘往右边缩
        }
    }
    return false;
}
int main(){
    int n,m;
    while (scanf("%d", &n) != EOF){
        for (int i = 0; i < n; ++i) {
            scanf("%d", &arr[i]);
        }
        // 排序
        sort(arr,arr+n);
        // 读取m个数据
        scanf("%d", &m);
        for (int i = 0; i < m; ++i) {
            int x;
            scanf("%d", &x);
            if(binarySearch(n,x)){
                printf("Yes\n");
            }
            else{
                printf("No\n");
            }
        }
    }
}

分析：

二分查找是最常用的查找方法，但我们首先要保证数组本身是有序的。每次通过中间位置mid，如果要查的比mid小，那么就缩到左区间继续查，即right = mid - 1;同理，如果要查的比mid大，那么就缩到右区间继续查，即left = mid + 1。（！！！注意出循环条件：left <= right）

代码示例2：

//find函数
#include <cstdio>
#include <map>
using namespace std;
int main(){
    map<int,int> findIndex;
    int m,n;
    int arr[101];
    while (scanf("%d",&n)  != EOF){
        for (int i = 0; i < n; ++i) {
            scanf("%d", &arr[i]);
            findIndex[arr[i]] = i;
            //将数组的元素作为键，数组元素的下标作为值，插入到map中
        }
        scanf("%d",&m);
        for (int i = 0; i < m; ++i) {
            int findNum;//待查找的元素
            scanf("%d", &findNum);
            //find函数会返回找到的那个元素的迭代器
            if(findIndex.find(findNum) == findIndex.end()){
                //在这个情况下，是没有找到元素
                printf("NO\n");
            }
            else{
                printf("YES\n");
            }
        }
    }
}

分析：

可以使用map解决查找问题，如下是map容器的原理图：（map的底层其实为二叉搜索树）

如果没有找到元素即指针迭代器找到了最后的后面，即findIndex.find(findNum) == findIndex.end()，此时打印NO。