挑战程序设计竞赛：0101抽签

最新推荐文章于 2024-06-25 21:19:32 发布

原创最新推荐文章于 2024-06-25 21:19:32 发布 · 797 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#枚举 #二分查找 #散列表 #挑战程序设计竞赛

收录于

ACM 同时被 2 个专栏收录

6 篇文章

订阅专栏

算法和数据结构

6 篇文章

订阅专栏

本文探讨了在给定整数集合中寻找四个数之和等于特定目标值的有效算法。提出了两种解决方案：一种是简单的暴力搜索，另一种是通过优化枚举结合二分搜索的方法来提高效率。

抽签

从大小为n整数集合A中每次抽一个元素x0记录后放回，要求判断抽四次和恰好为m方案是否存在？

测试数据：

– 输入 n=3 A={1,3,5} m=9 输出Yes 因为 1+1+3+5=9所以存在
– 输入 n=3 A={1,3,5} m=5 输出No 因为不存在

方法一：暴力 O(n^4)

bool solve(int n, int m, vector<int> &k) {
    for (auto a:k)
        for (auto b:k)
            for (auto c:k)
                for (auto d:k) {
                    if (a + b + c + d == m)
                        return true;
                }
    return false;
}

方法二：优化枚举+二分搜索O(n^2log(n))

bool solve(int n, int m, vector<int> &k) {
    vector<int> k2;
    for (size_t i = 0; i < k.size(); i++)
        for (size_t j = 0; j < k.size(); j++) {
            k2.push_back(k[i] + k[j]);
        }
    sort(k2.begin(), k2.end());
    for (size_t i = 0; i < k2.size(); i++) {
        if (binary_search(k2.begin(), k2.end(), m - k2[i]))
            return true;
    }
    return false;
}