【LeetCode】227、基本计算器 II

【LeetCode】227、基本计算器 II

文章目录

一、栈

1.1 栈

go 复制代码
// go
func calculate(s string) int {
	// 需要解析字符串 里的每个 token
	// 本题不涉及括号, 只有+-*/, 所以涉及优先级: */的优先级比+-高
	// 为了处理优先级, 需要用栈: 即先遇到的+-都依次放入栈, 当遇到*/时优先一次性计算完, 再入栈
	// 最终栈内只有+-两种符号, 不存在优先级的问题, 则把栈内元素依次+-即可

	// 因为计算, 都是 "数" "符号" "数" 计算, 如 1 + 2, 3 / 5, 所以用两个栈: 数字栈和符号栈存储

	// 示例1:
	// 数字栈 [],  符号栈 []
	// 数字栈 [3], 符号栈 [+]
	// 数字栈 [3,2], 符号栈 [+,*]
	// 待入栈的2+, 遇到栈顶的2*里的乘号, 则需先pop(2*), 再计算(2*2=4), 再push(4+)
	// 数字栈 [3,4], 符号栈 [+,+]
	// 此时已遍历完, 站内只有+-了, 直接计算即可: 3+4+ = 7

	// 其中数字可能超过一位, 例如 123, 4563 等, 则需 cur 初值为 0, 并 cur = cur * 10 + (c - 'a') 来把字符串转为数字

	nums := []int{} // 数字栈
	ops := []byte{} // 符号栈
	cur := 0        // 遍历到的数字
	for _, c := range []byte(s + "+") { // 补全 "+" 是为了最后一对儿 num 和 op
		if c == ' ' { continue } // 不处理空格
		if c >= '0' && c <= '9' { // 遇到某位数字则 保留多位数字
			cur = cur*10 + int(c-'0') // 字符串转数字, 支持多位数字
		} else if l := len(nums); l == 0 || ops[l-1] == '+' || ops[l-1] == '-' { // 数字栈为空, 或, 栈顶为 +-, 则压栈
			nums = append(nums, cur); cur = 0  // 数字压栈, 压栈后, 清空当前数字
			ops = append(ops, c) // 符号压栈
		} else { // 遇到 */ 则 pop(), 并计算, 并 push()
			n := len(nums); topnum := nums[n-1]; nums = nums[:n-1] // nums.pop()
			o := len(ops); topop := ops[o-1]; ops = ops[:o-1] // ops.pop()

			tmp := 0 // 高优先级的 */ 的计算结果
			if topop == '*' {
				tmp = topnum * cur
			} else {
				tmp = topnum / cur
			}
            cur = 0

			nums = append(nums, tmp)
			ops = append(ops, c)
		}
	}
	return compute(nums, ops)
}

// 只有+-符号了
func compute(nums []int, ops []byte) int {
	n := len(nums)
    ans := nums[0]
	for i := 1; i < n; i++ {
		op := ops[i-1]
		if op == '+' {
			ans += nums[i]
		} else {
			ans -= nums[i]
		}
	}
	return ans
}

参考左神视频讲解

二、多语言解法

C p p / G o / P y t h o n / R u s t / J s / T s Cpp/Go/Python/Rust/Js/Ts Cpp/Go/Python/Rust/Js/Ts

cpp 复制代码
// cpp
go 复制代码
// go 同上
python 复制代码
# python
rust 复制代码
// rust
js 复制代码
// js
ts 复制代码
// ts
相关推荐
森焱森1 小时前
无人机三轴稳定化控制(1)____飞机的稳定控制逻辑
c语言·单片机·算法·无人机
循环过三天1 小时前
3-1 PID算法改进(积分部分)
笔记·stm32·单片机·学习·算法·pid
闪电麦坤951 小时前
数据结构:二维数组(2D Arrays)
数据结构·算法
凌肖战1 小时前
力扣网C语言编程题:快慢指针来解决 “寻找重复数”
c语言·算法·leetcode
埃菲尔铁塔_CV算法2 小时前
基于 TOF 图像高频信息恢复 RGB 图像的原理、应用与实现
人工智能·深度学习·数码相机·算法·目标检测·计算机视觉
NAGNIP3 小时前
一文搞懂FlashAttention怎么提升速度的?
人工智能·算法
Codebee3 小时前
OneCode图生代码技术深度解析:从可视化设计到注解驱动实现的全链路架构
css·人工智能·算法
刘大猫263 小时前
Datax安装及基本使用
java·人工智能·算法
Gyoku Mint6 小时前
深度学习×第4卷:Pytorch实战——她第一次用张量去拟合你的轨迹
人工智能·pytorch·python·深度学习·神经网络·算法·聚类