【每日一题】CF1692 F. 3SUM | 枚举 | 简单

solego

已于 2024-04-08 13:53:45 修改

阅读量734

点赞数 13

分类专栏：算法竞赛文章标签：算法

于 2024-04-08 13:50:20 首次发布

本文链接：https://blog.csdn.net/weixin_43900869/article/details/137505141

版权

算法竞赛专栏收录该内容

61 篇文章 0 订阅

订阅专栏

文章讲述了如何解决一个关于整数数组的问题，即找出是否存在三个下标不同的元素，其个位数相加等于3。方法是利用一个长度为10的数组记录每个个位数的出现次数，通过枚举并更新计数来判断是否能找到符合条件的组合。

摘要由CSDN通过智能技术生成

题目内容

原题链接

给定一个长度为 $n$ 的数组 $a$ ，问是否可以选择三个下标不同的元素 $a_i,a_j,a_k$ ，满足 $a_i+a_j+a_k$ 的个位数为 $3$ 。

数据范围

$3\leq n\leq 2\cdot 10^5$
$1\leq a_i\leq 10^9$

题解

和的个位数为 $3$ 只与每个数的个位有关，所以用一个长度为 $10$ 的数组保存每个数的数量。

不妨假定三个数中

$a_i$ 的个位数小于等于 $a_j$ 的个位数
$a_j$ 的个位数小于等于 $a_k$ 的个位数

如果我们选择了两个或三个个位数相同的数，需要保证这个个位数对应数的数量足够，所以可以在选择之后将其数量减 1 ，继续枚举之后的数，在本轮枚举结束后，将其数量加 1 恢复回来。

时间复杂度： $O (10 * 10 * 10)$

代码

#include <bits/stdc++.h>
using namespace std;

void solve() {
    int n;
    cin >> n;

    vector<int> cnt(10);
    vector<int> a(n);
    for (int i = 0; i < n; ++i) {
        cin >> a[i];
        cnt[a[i] % 10] += 1;
    }

    for (int i = 0; i < 10; ++i) {
        if (!cnt[i]) continue;
        cnt[i] -= 1;
        for (int j = i; j < 10; ++j) {
            if (!cnt[j]) continue;
            cnt[j] -= 1;
            for (int k = j; k < 10; ++k) {
                if (!cnt[k]) continue;
                cnt[k] -= 1;
                if ((i + j + k) % 10 == 3)  {
                    cout << "YES\n";
                    return;
                }
                cnt[k] += 1;
            }
            cnt[j] += 1;
        }
        cnt[i] += 1;
    }
    cout << "NO\n";
}

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

    int T = 1;
    cin >> T;
    while (T--) {
        solve();
    }

    return 0;
}