《算法笔记》4.5小节——算法初步-＞二分 问题 A: 找x

问题 A: 找x

时间限制 : 1.000 sec 内存限制 : 32 MB

题目描述

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

输入

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

输出

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

样例输入

4
1 2 3 4
3

样例输出

2

输入数据不保证有序，这代码凑巧能过而已

#include<iostream>
#include <algorithm>


using namespace std;
int a[210];

int find(int l, int r, int x) {
    while (l <= r) {
        int mid = (l + r) >> 1;
        if (a[mid] > x) r = mid - 1;
        else if (a[mid] < x)
            l = mid + 1;
        else
            return mid;
    }
    return -1;
}

int main() {
    int n;
    while (cin >> n) {
        int x;
        for (int i = 0; i < n; ++i)
            cin >> a[i];
        cin >> x;
        cout << find(0, n - 1, x) << endl;
    }
    return 0;
}

用map实现效率更高

#include<iostream>
#include <algorithm>
#include <cstring>
#include <unordered_map>


using namespace std;
int a[210];
int n;


int main() {
    while (cin >> n) {
        int x;
        memset(a, 0, sizeof a);
        unordered_map<int, int> map;
        for (int i = 0; i < n; ++i) {
            cin >> a[i];
            map.insert({a[i], i});
        }
        cin >> x;
        if (map.find(x) != map.end())
            cout << map.find(x)->second << endl;
        else
            cout << "-1" << endl;
    }
    return 0;
}