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;
} 

总结

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

相关推荐
空白到白3 分钟前
机器学习-聚类
人工智能·算法·机器学习·聚类
小马学嵌入式~6 分钟前
嵌入式 SQLite 数据库开发笔记
linux·c语言·数据库·笔记·sql·学习·sqlite
索迪迈科技9 分钟前
java后端工程师进修ing(研一版 || day40)
java·开发语言·学习·算法
hour_go23 分钟前
用户态与内核态的深度解析:安全、效率与优化之道
笔记·操作系统
zzzsde23 分钟前
【数据结构】队列
数据结构·算法
芒克芒克32 分钟前
LeetCode 面试经典 150 题:删除有序数组中的重复项(双指针思想解法详解)
算法
青 .33 分钟前
数据结构---二叉搜索树的实现
c语言·网络·数据结构·算法·链表
Zz_waiting.34 分钟前
Javaweb - 14.6 - Vue3 数据交互 Axios
开发语言·前端·javascript·vue·axios
萌新小码农‍36 分钟前
Java分页 Element—UI
java·开发语言·ui
Tiger_shl1 小时前
【.Net技术栈梳理】03-核心框架与运行时(异常处理)
开发语言·.net