专栏首页SnailTyanLeetcode 726. Number of Atoms

Leetcode 726. Number of Atoms

文章作者:Tyan 博客:noahsnail.com | CSDN | 简书

1. Description

2. Solution

解析:这道题还有优化的空间,这样写主要是逻辑清晰。1. 把元素(多个字母)、数字(多个数字字符)、左右括号拆分开;2. 计算元素的个数,如果元素后没有数字,则添加数字1作为元素个数;当碰到右括号时,查找其对应的左括号,并将其中的元素个数乘以括号后的数字,其后没数字,则默认乘以1;3. 统计元素个数,相同元素个数相加;4. 排序字典,按元素字母排序;5. 构造返回结果字符串。

  • Version 1
class Solution:
    def countOfAtoms(self, formula):
        stat = {}
        stack = []
        parts = []

        # Split string by alpha, number, '(', ')'
        for index, ch in enumerate(formula):
            if ch.isupper():
                parts.append(ch)
            elif ch.islower():
                parts[-1] += ch
            elif ch.isdigit():
                if formula[index - 1].isdigit():
                    parts[-1] += ch
                else:
                    parts.append(ch)
            else:
                parts.append(ch)

        # Calculate the number of atom and remove '(', ')'
        stack = []
        for index, part in enumerate(parts):
            if part.isalpha():
                if index + 1 == len(parts) or not parts[index + 1].isdigit():
                    stack.append(part)
                    stack.append(1)
                else:
                    stack.append(part)
            elif part.isdigit() and parts[index - 1] != ')':
                    stack.append(int(part))
            elif part == '(':
                stack.append(part)
            elif part == ')':
                if index + 1 < len(parts) and parts[index + 1].isdigit():
                    multiplier = int(parts[index + 1])
                else:
                    multiplier = 1

                i = len(stack) - 1
                while stack[i] != '(':
                    if isinstance(stack[i], int):
                        stack[i] = stack[i] * multiplier
                    i -= 1
                stack.pop(i)

        # Stat the number of atoms
        for i in range(0, len(stack), 2):
            if stack[i] in stat:
                stat[stack[i]] += stack[i + 1]
            else:
                stat[stack[i]] = stack[i + 1]
        stat = sorted(stat.items(), key=lambda item: item[0])
        result = ''
        for key, value in stat:
            if value > 1:
                result = result + key + str(value)
            else:
                result = result + key
        return result

Reference

  1. https://leetcode.com/problems/number-of-atoms/

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

我来说两句

0 条评论
登录 后参与评论

相关文章

  • LWC 58:726. Number of Atoms

    LWC 58:726. Number of Atoms 传送门:726. Number of Atoms Problem: Given a chemical ...

    用户1147447
  • DOCK-4-生成 Grid

    围绕活性位点创建一个box 使用指令:showbox < box.in box.in的文件格式为:

    FindKey
  • 【独家推送】GoogLeNet构建技术分析因子的模式识别基于TensorFlow

    系列文章(点击即可查看) 机器学习该如何应用到量化投资系列(一) 机器学习该如何应用到量化投资系列(二) 机器学习该如何应用到量化投资系列(三) 机器学习该如...

    量化投资与机器学习微信公众号
  • Julia-BioStructures

    BioStructures提供了读取,写入和操纵大分子结构(蛋白质)的功能。可以将Protein Data Bank(PDB),mmCIF和MMTF格式的文件读...

    FindKey
  • LeetCode <BT> 200. Number of Islands

    Given a 2d grid map of '1's (land) and '0's (water), count the number of islands...

    大学里的混子
  • Leetcode 200. Number of Islands

    版权声明:博客文章都是作者辛苦整理的,转载请注明出处,谢谢! https://blog.cs...

    Tyan
  • Leetcode 233. Number of Digit One

    Given an integer n, count the total number of digit 1 appearing in all non-nega...

    triplebee
  • Leetcode 191 Number of 1 Bits

    Write a function that takes an unsigned integer and returns the number of ’1' b...

    triplebee
  • LeetCode 191. Number of 1 Bits

    Write a function that takes an unsigned integer and returns the number of '1' bi...

    大学里的混子

扫码关注云+社区

领取腾讯云代金券