两数之和变形,CF 1278C - Berry Jam

目录

一、题目

1、题目描述

2、输入输出

2.1输入

2.2输出

3、原题链接

二、解题报告

1、思路分析

2、复杂度

3、代码详解


一、题目

1、题目描述

2、输入输出

2.1输入
2.2输出

3、原题链接

1278C - Berry Jam


二、解题报告

1、思路分析

考虑保留的前后缀满足什么特点?

前缀中 1的数目 - 2的数目 = 后缀中 2的数目 - 1的数目

那么我们发现这就是一个两数之和问题

我们哈希表预处理最靠右 前缀中 1的数目 - 2的数目 的位置

然后倒着遍历后缀,计数维护最长前后缀长度即可

2、复杂度

时间复杂度: O(NlogN)(可以将下标映射从而避免map的访问开销)空间复杂度:O(N)

3、代码详解

复制代码
 ​
cpp 复制代码
#include <bits/stdc++.h>

using i64 = long long;
using i32 = unsigned int;
using u64 = unsigned long long;
using i128 = __int128;

constexpr int inf32 = 1E9 + 7;
constexpr i64 inf64 = 1E18 + 7;
constexpr int P = 998'244'353;

void solve() {
	int n;
	std::cin >> n;
	
	int m = n * 2;

	std::vector<int> a(m);
	std::map<int, int> pre, suf;

	for (int i = 0; i < m; ++ i) std::cin >> a[i];

	for (int i = 0, f = 0; i < n; ++ i) {
		f += (a[i] & 1 ? 1 : -1);
		pre[f] = i;
	}

	int res = m;

	for (int i = m - 1, f = 0; i >= n; -- i) {
		f += (a[i] & 1 ? -1 : 1);
		if (pre.contains(f)) {
			res = std::min(res, i - pre[f] - 1);
		}
		if (f == 0)
			res = std::min(res, i);
	}

	if (pre.contains(0))
		res = std::min(res, m - pre[0] - 1);

	std::cout << res << '\n';
}

auto FIO = []{
	std::ios::sync_with_stdio(false);
	std::cin.tie(nullptr);
	std::cout.tie(nullptr);
	return 0;
}();

int main () {
	#ifdef DEBUG
		freopen("in.txt", "r", stdin);
		freopen("out.txt", "w", stdout);
	#endif
	
	int T = 1;
	std::cin >> T;
	while (T --) {
		solve();
	}

	return 0;
}
相关推荐
小O的算法实验室17 小时前
2022年ASOC SCI2区TOP,基于竞争与合作策略的金字塔粒子群算法PPSO,深度解析+性能实测,深度解析+性能实测
算法·论文复现·智能算法·智能算法改进
南莺莺17 小时前
邻接矩阵的基本操作
数据结构·算法··邻接矩阵
微波仿真18 小时前
实现多通道ADC多次测量取平均值,使用DMA
算法
余俊晖18 小时前
多模态文档理解视觉token剪枝思路
人工智能·算法·剪枝·多模态
aramae19 小时前
详细分析平衡树--红黑树(万字长文/图文详解)
开发语言·数据结构·c++·笔记·算法
再卷也是菜19 小时前
C++篇(13)计算器实现
c++·算法
CHEN5_0219 小时前
【leetcode100】和为k的子数组(两种解法)
java·数据结构·算法
Codeking__19 小时前
DFS算法原理及其模板
算法·深度优先·图论
Victory_orsh19 小时前
“自然搞懂”深度学习系列(基于Pytorch架构)——01初入茅庐
人工智能·pytorch·python·深度学习·算法·机器学习
88号技师19 小时前
2025年8月SCI-汉尼拔·巴卡优化算法Hannibal Barca optimizer-附Matlab免费代码
开发语言·人工智能·算法·数学建模·matlab·优化算法