Nim早茶之使用栈实现计算器

我们这次实现的命令行计算器,支持加减乘除、括号、浮点数、负数,以及查看历史和退出功能。

主要的思路:read - parse - print - loop。

read 阶段是指从读取用户在提示符(cal> )之后输入的字符串。

parse 阶段包括:将用户输入的字符串分割成单个对象比如:符号 +,或者数字 1.2 。其次将中缀表达式转化后缀表达式,接下来计算后缀表达式的数值。

print 阶段指的是将 parse 阶段的数值打印在终端上。

loop 阶段是指重复上述操作,除非用户输入 exit 符号或者 Ctrl + C 终止程序运行。

实现效果图:

55.gif

为了实现后面的中缀、后缀表达式,我们需要借助栈这种数据结构。

栈是一种先进后出的数据结构,函数调用时就会使用栈来保存当前程序的地址及其他信息,前面的深度优先搜索就是基于栈实现的。

之前写了一篇文章介绍了栈的用法,下面简单说明栈的使用:

push 函数在栈的顶部添加元素,pop 函数删除栈顶的元素,top 函数查看栈顶函数。

# nim编程
type Stack*[T] = ref object
  container*: seq[T] 


proc len*[T](s: Stack[T]): int =
    s.container.len

proc push*[T](s: Stack[T], elem: T) = 
  s.container.add(elem)

proc pop*[T](s: Stack[T]): T =
  s.container.pop()

proc top*[T](s: Stack[T]): T =
  if s.len > 0:
    result = s.container[s.len - 1]

分词

分词是 parse 阶段的重要部分。

  • 我们定义符号 + - * / ( )。

  • 将这些字符串中的这些符号替换为:空格 + 符号 + 空格

+ => 空格+空格
  • 将字符串按空格分割,得到最终的分词

经过上述操作,就可以实现对正数以及操作符的分词功能。

# nim 编程
iterator token*(expressions: string): string = 
  var sentences: string = expressions
  for symbol in symbols:
    # 替换字符串
    sentences = sentences.replace(symbol, fmt" {symbol} ") 
  var temp: seq[string]
  # 分割字符串
  for sentence in sentences.split:
    if sentence != "":
      temp.add(sentence)

为了实现对负数的分词,我们需要迭代分词序列,将‘-’ 与后面的数字转换成一个整体。我们有两种情况将 - 视作负号,第一种情况,负号位于字符串的开头;第二种情况,负号前一个符号为左括号。

  var target: seq[string] = temp
  for (i, item) in temp.pairs:
    if item == "-": 
      # 两种情况下,当前符号设置为 ""
      # 下一个符号设置为负数
      if (i-1) < 0 or temp[i-1] == "(": 
        target[i] = ""
        target[i+1] = fmt"-{target[i+1]}"
    # 若字符串不为空,打印字符串
    if target[i] != "":
      yield target[i]

中缀表达式

中缀表达式就是我们写的各种表达式,如 3 + 2 * 4。我们希望将其转化为后缀表达式,示例如下:

中缀表达式 ==> 1 + 5 * 4 + (3 * 2 + 4) * 6
后缀表达式 ==> 1 5 4 * + 3 2 * 4 + 6 * + 

我们需要准备一个栈 stack ,用于临时存放操作符;准备一个序列 seq,用来存放输出结果,也就是后缀表达式。

转化规则:

迭代分词序列,遇到数字,就将数字输出到序列 seq 中。

定义各个符号优先级,其中 乘除 > 加减 > 右括号。比较当前符号与栈顶元素的优先级,如果当前元素的优先级小于等于栈顶元素的优先级,就弹出栈顶元素,输出到 seq 序列中,然后将当前元素压入栈中。如果当前元素的优先级大于栈顶元素的优先级,或者当前元素为右括号,我们直接将当前元素压入栈中。

如果遇到左括号,我们就弹出栈中元素,并输出到 seq 序列中。其中,左右括号并不输出到序列中。

最后,将栈中剩余的符号输出到 seq 序列中。

# nim编程
# 迭代 1 + 5 * 4 + (3 * 2 + 4) * 6
# 读入 1 并输出到 seq 中,将 + 压入栈中
# 读入 5 并输出到 seq 中
stack ==> +
seq   ==> 1 5
# 读入 *, * 优先级 > + 优先级
# 将 * 压入栈中, 读入 4 并输出
stack ==> + *
seq   ==> 1 5 4
# 读入 +, + 优先级 < * 优先级
# 将 * 输出到 seq 序列中
# + 优先级 <= + 优先级,+ 输出
# 将 + 压入栈中
stack ==> +
seq   ==> 1 5 4 * +
# 读入 (, 压入栈中,读入 3,压入栈中
stack ==> + (
seq   ==> 1 5 4 * + 3
# 读入 *, 压入栈中,读入 2,压入栈中
stack ==> + ( *
seq   ==> 1 5 4 * + 3 2
# 读入 +,* 输出,+ 压入栈中, 4 输出
stack ==> + ( +
seq   ==> 1 5 4 * + 3 2 * 4
# 读入 ),+ 输出,弹出 (
stack ==> +
seq   ==> 1 5 4 * + 3 2 * 4 +
# 读入 *, 压入栈中, 6 输出
stack ==> + *
seq   ==> 1 5 4 * + 3 2 * 4 + 6
# 将栈中剩余符号输出
stack ==> 
seq   ==> 1 5 4 * + 3 2 * 4 + 6 * +

代码如下:

# nim编程
proc parseInfix(expressions: string): seq[string] = 
  var s: Stack[string] = Stack[string](container: @[])
  var target: seq[string] = @[]
  for expression in token(expressions):
    let flag = (expression in symbols)
    if not isFloat(expression) and not flag:
      raise newException(ValueError, "error")

    if isFloat(expression):
      target.add(expression)

    elif expression == ")":
      while s.top != "(": 
        target.add(s.pop)
      discard s.pop

    elif expression == "(":
      s.push(expression)

    elif s.top != "" and comparePriority(expression, s.top):
      target.add(s.pop)
      while s.top != "" and comparePriority(expression, s.top):
        target.add(s.pop)
      s.push(expression)
    else:
      s.push(expression)

  while s.len != 0:
    target.add(s.pop)
  result = target

后缀表达式

后缀表达式的规则:遇到数值就压入栈中;遇到符号,就从栈中弹出两个元素,计算两个元素运算结果,将得到的结果压入栈中。

# nim编程
proc parseSufix(expressions: seq[string]): float =
  var target: float
  var s: Stack[string] = Stack[string](container: @[])
  for expression in expressions:
    if isFloat(expression):
      s.push(expression)
    else:
      var elem2 = parseFloat(s.pop)
      var elem1 = parseFloat(s.pop)
      target = evaluate(expression, elem1, elem2)
      s.push(fmt"{target}")
  if s.len > 0:
    target = parseFloat(s.pop)
  result = target 

总结

接下来还有添加查看历史和退出功能以及整个程序的组成,关注微信公众号nim编程在后台回复 2019813获取源代码及 exe 文件。
Nim 个人博客:https://tea.nim-cn.com/

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 218,204评论 6 506
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 93,091评论 3 395
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 164,548评论 0 354
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,657评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,689评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,554评论 1 305
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,302评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 39,216评论 0 276
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,661评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,851评论 3 336
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,977评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,697评论 5 347
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,306评论 3 330
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,898评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 33,019评论 1 270
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 48,138评论 3 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,927评论 2 355

推荐阅读更多精彩内容