AT_abc351_c [ABC351C] Merge the balls 题解

题目传送门

题目大意

你有一个空序列和 N N N 个球。第 i i i 个球 ( 1 ≤ i ≤ N ) (1 \leq i \leq N) (1≤i≤N) 的大小是 2 A i 2^{A_i} 2Ai。

计算 N N N 操作后序列中剩余的球的个数。

你将进行 N N N 次运算。

在第 i i i 次操作中,你将第 i i i 个球添加到序列的最右端,然后重复下面的步骤:

  1. 如果序列中只有一个或更少的球,则结束操作。
  2. 如果序列中最右边的球和第二右边的球大小不同,则结束操作。
  3. 如果序列中最右边的球和第二右边的球大小相同,则拿出这两个球,并在序列的最右端添加一个新球,其大小等于移除的两个球的大小之和。然后回到步骤 1,重复上述过程。

计算 N N N 操作后序列中剩余的球的个数。

解题思路

我们拿到题目一看,可以直接模拟,算法时间复杂度 O ( n ) O(n) O(n)。

首先定义一个 stack,为什么要定义一个栈呢?因为我们可以发现,每次操作都是从序列的最右端取出球,而不关最左端的球的事,所以我们可以将序列的最右端看成栈顶,而最左端可以看做是栈底,于是我们只需要在每次操作时,先将第 i i i 个球入栈,再判断栈的元素个数,然后连续两次出栈,判断两球大小是否相同,最后确定是否将新球入栈即可。

大体思路如上所述,但是还有一个地方困住了一些选手,就是如题所述的第 i i i 个球的大小为 2 A i 2^{A_i} 2Ai,但是 0 ≤ A i ≤ 1 0 9 0\le A_i\le10^9 0≤Ai≤109,那我们肯定不能直接算出 2 A i 2^{A_i} 2Ai 的值,因为存不下这么大的数。我们不难想到在每次操作中只计算 A i A_i Ai 的值,但是这怎么计算呢?我们又得从题目进行分析,题目说了只有在最右端的两球大小相同时才会添加一个大小为拿出的球的大小之和的球,也就是说我们每次计算时,只需要计算 2 × 2 B k 2\times 2^{B_k} 2×2Bk (这里令 B k B_k Bk 为拿走的其中一个球的大小)的值就可以了。而 2 × 2 B k 2\times 2^{B_k} 2×2Bk 就等于 2 B k + 1 2^{B_k+1} 2Bk+1,所以在添加球的时候我们只需要入栈 B k + 1 B_k + 1 Bk+1 即可。

CODE:

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define int long long
stack<int> s;
int a[200010];
signed main() {
	ios::sync_with_stdio(false);
	ios_base::sync_with_stdio(false);
	cin.tie(0), cout.tie(0);
	int n;
	cin >> n;
	for (int i = 1; i <= n; i++)
		cin >> a[i];
	for (int i = 1; i <= n; i++) {
		s.push(a[i]);
		while (1) {
			if (s.size() <= 1) {
				break;
			}
			int k = s.top();
			s.pop();
			int l = s.top();
			s.pop();
			if (k != l) {
				s.push(l);
				s.push(k);
				break;
			}
			s.push(l + 1);
		}
	}
	cout << s.size();
	return 0;
} 

总结

这道题目主要考察了大家能不能将题目描述转化为栈的操作,总体来说较为简单。

相关推荐
猫头虎11 小时前
如何解决 pip install -r requirements.txt extras 语法 ‘package[extra’ 缺少 ‘]’ 解析失败问题
开发语言·python·开源·beautifulsoup·virtualenv·pandas·pip
zhangfeng113311 小时前
R语言 读取tsv的三种方法 ,带有注释的tsv文件
开发语言·r语言·生物信息
eqwaak011 小时前
动态图表导出与视频生成:精通Matplotlib Animation与FFmpeg
开发语言·python·ffmpeg·音视频·matplotlib
刘新明198911 小时前
Frida辅助分析OLLVM虚假控制流程(下)
java·开发语言·前端
澄澈i11 小时前
CMake学习篇[3]---CMake进阶+嵌套CMakeLists+多层级关系
c++·学习·cmake
如意猴11 小时前
双向链表----“双轨联动,高效运行” (第九讲)
数据结构·链表
搂鱼11451411 小时前
GJOI 10.7/10.8 题解
算法
第二只羽毛12 小时前
重载和继承的实践
java·开发语言
Django强哥12 小时前
JSON Schema Draft-07 详细解析
javascript·算法·代码规范
AndrewHZ12 小时前
【图像处理基石】GIS图像处理入门:4个核心算法与Python实现(附完整代码)
图像处理·python·算法·计算机视觉·gis·cv·地理信息系统