判断无序数组中是否存在两个数之和为m

版权声明:所有的博客都是个人笔记,交流可以留言。未经允许,谢绝转载。。。 https://blog.csdn.net/qq_35976351/article/details/89478004

题目描述

给定一个长度是n整数的无序序列,然后给定一个整数m,判定是序列中是否有两个数的和是m。注意,数列是无序的,数据可能有重复的。

解题思路

需要利用哈希进行解题,这样保证复杂度是 O ( n ) O(n) 。 注意一个特别情况,加入数据序列是{1, 3, 5},数据是6,那么3的情况下会出问题。因此解决方案是:如果m是偶数,而且当前的数是m/2而且只出现了一次,那么肯定不是。

AC代码

#include <iostream>
#include <vector>
#include <unordered_map>

int main() {
    std::vector<int> vec;
    int m, n;
    std::cin >> n >> m;
    bool even = (m % 2 == 0);
    for (int i = 0; i < n; ++i) {
        int x;
        std::cin >> x;
        vec.emplace_back(x);
    }

    std::unordered_map<int, int> nums;
    for (int i = 0; i < n; ++i) {
        int x = m - vec[i];
        auto it = nums.find(x);
        if (it == nums.end()) {
            nums.emplace(std::make_pair(x, 1));
        } else {
            it->second += 1;
        }
    }

    bool tag = false;
    for (int i = 0; i < n; ++i) {
        auto it = nums.find(m - vec[i]);
        if (it != nums.end()){
            if (even && it->first == m / 2 && it->second == 1) {  // 偶数情况出现一半
                continue;
            }
            tag = true;
            break;
        }
    }

    if (tag) {
        std::cout << "True\n";
    } else {
        std::cout << "False\n";
    }

    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_35976351/article/details/89478004