[二分查找/哈希查找]P2249 【深基13.例1】查找 - 洛谷

P2249 【深基 13. 例 1】查找 - 洛谷
https://www.luogu.com.cn/problem/solution/P2249

题目来源

洛谷

题目内容

【深基13.例1】查找

题目描述

输入 $n$ 个不超过 $10^9$ 的单调不减的（就是后面的数字不小于前面的数字）非负整数 $a_1,a_2,\dots ,a_n$，然后进行 $m$ 次询问。对于每次询问，给出一个整数 $q$，要求输出这个数字在序列中第一次出现的编号，如果没有找到的话输出 $-1$ 。

输入格式

第一行 $2$ 个整数 $n$ 和 $m$，表示数字个数和询问次数。

第二行 $n$ 个整数，表示这些待查询的数字。

第三行 $m$ 个整数，表示询问这些数字的编号，从 $1$ 开始编号。

输出格式

输出一行，$m$ 个整数，以空格隔开，表示答案。

样例 #1

样例输入 #1

11 3
1 3 3 3 5 7 9 11 13 15 15
1 3 6

样例输出 #1

1 2 -1

提示

数据保证，$1\le n\le 10^6$，$0\le a_i,q\le 10^9$，$1\le m\le 10^5$

本题输入输出量较大，请使用较快的 IO 方式。

知识点

1. STL + lower_bound ()
2. 哈希 + 查找

题目思路

看数据量，要用比$O(n^2)$ 快的算法

1. 哈希查找的速度快，通常可以在 O (1) 的时间复杂度下完成查找。
2. lower_bound () ，它通常使用二分查找来实现，因此其时间复杂度为 $O(logn)$，其中 n 是容器中元素的数量。

AC代码

解法一: STL + lower_bound()

//
// Created by Jisam on 28/09/2024 10:48 PM.
// Solution of  查找
// 2024-09-28 23:09:06 AC 100 STL + lower_bound() 
#include <bits/stdc++.h>

using namespace std;
using i64 = long long;
using u64 = unsigned long long;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    // 读取整数n和m的值
    int n, m;
    cin >> n >> m;
    
    // 初始化一个大小为n的整数向量a，并读取每个元素的值
    vector<int> a(n);
    for(auto & x : a) cin >> x;
    
    // 重复m次的操作
    while (m --){
        // 读取查询值q
        int q;
        cin >> q;
       
        // 使用二分查找确定q在向量a中首次出现的位置
        int ans = lower_bound(a.begin(),a.end(),q) - a.begin();
        
        // 输出结果：如果q存在于向量a中，则输出其位置（从1开始计数）；否则输出-1
        cout <<( a[ans] == q ? ans + 1 : -1) << " ";
    }
    return 0
}

解法二： 哈希 + 查找

//
// Created by Jisam on 28/09/2024 10:48 PM.
// Solution of  查找
// 2024-09-28 22:59:57 AC 100  哈希 + 查找

#include <bits/stdc++.h>

using namespace std;
using i64 = long long;
using u64 = unsigned long long;

int main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    // 读取整数n和m的值
    int n, m;
    cin >> n >> m;

    // 初始化一个无序映射maps来存储每个元素及其首次出现的位置
    unordered_map<int, long long> maps;

    // 读取n个输入值并更新maps
    int in = 0;
    for (int i = 0; i < n; i++) {
        cin >> in;
        if (maps[in] == 0) {
            maps[in] = i + 1;
        }
    }

    // 重复m次的操作
    int x = 0;
    for (int i = 0; i < m; i++) {
        cin >> x;

        // 如果x存在于maps中，则输出其位置（从1开始计数）；否则输出-1
        if (maps[x]) {
            cout << maps[x] << " ";
        } else {
            cout << -1 << " ";
        }
    }

    cout << endl;
    return 0
}