F. 3SUM

最新推荐文章于 2024-09-14 18:59:16 发布

Angindem

最新推荐文章于 2024-09-14 18:59:16 发布

阅读量62

点赞数

分类专栏：玩转上号CF“游戏” 算法错题本文章标签： c语言 c++

本文链接：https://blog.csdn.net/hacker_51/article/details/132524970

版权

算法错题本同时被 2 个专栏收录

71 篇文章 0 订阅

订阅专栏

玩转上号CF“游戏”

38 篇文章 0 订阅

订阅专栏

题目：

样例：

输入
`6 4 20 22 19 84 4 1 11 1 2022 4 1100 1100 1100 1111 5 12 34 56 78 90 4 1 9 8 4 6 16 38 94 25 18 99`

输出
YES YES NO NO YES YES

解题思路：

这道题很好玩，找出任意三个元素，相加有没有结果个位数为 3 的三个元素

这里我们三重循环个位数 0~9相加即可，输入的时候统计每个元素的相应的个位数，该个位数共有多少个，然后暴力枚举一遍可以了

代码详解如下：

#include <iostream>
#include <unordered_map>
#define endl '\n'
#define YES puts("YES")
#define NO puts("NO")
#define umap unordered_map
#pragma GCC optimize(3,"Ofast","inline")
#define ___G std::ios::sync_with_stdio(false),cin.tie(0), cout.tie(0)
using namespace std;
const int N = 2e6 + 10;

int n, sum;
umap<int, int>r;	// 用于记录我们数组元素中个位数出现的次数

inline void solve()
{
	//清空之前的记录
	r.clear();

	cin >> n;
	for (int i = 0, x; i < n; ++i)
	{
		cin >> x;

		// 由于结果只要个位数是 3，所以只有个位数有关系
		// 取出个位数
		x %= 10;

		// 记录好该数值有多少个
		r[x]++;

		// 这里来个小优化，记录所有个位数的总和
		sum += x;
	}

	// 如果所有个数都不可能超过 3 说明是 NO
	if (sum < 3)
	{
		NO;
		return ;
	}

	sum = 0;	// 初始化所有总和


	// 暴力枚举所有可能的个位数
	for (int i = 0; i < 10; ++i)
	{
		for (int j = i; j < 10; ++j)
		{
			for (int k = j; k < 10; ++k)
			{
				r[i]--, r[j]--, r[k]--;	// 取出我们枚举的数值
				if ((i + j + k) % 10 == 3 && r[i] >= 0 && r[j] >= 0 && r[k] >= 0)
				{
					YES;
					return ;
				}

				// 由于不符合，所以放回去，恢复我们取出来
				r[i]++, r[j]++, r[k]++;
			}
		}
	}
	NO;
}


int main()
{
//	freopen("a.txt", "r", stdin);
	___G;
	int _t = 1;
	cin >> _t;
	while (_t--)
	{
		solve();
	}

	return 0;
}