面试题 - 京东(笔试). 有机化合物

题目描述

小明最近在学习《有机化学》。在化学中，有机化合物简称为有机物，是指含碳元素的化合物，其组成元素除碳外，通常还含有氢、氮、硫、磷、卤素等元素。
小明想写个小程序来计算简单有机化合物的分子量（相对分子量），假设需要计算分子量的有机物中只包含C、H、O、N四种元素.
其相对原子质量分别如下：
C:12、H:1、O:16、N:14

例如某有机物的分子式为：C4H9OH，其相对分子量为：12×4 + 1×9 + 16 + 1 = 74

输入描述

单组输入。
输入一行，包含一个只包含字母C、H、O、N和数字的分子式，如果字母后面没有数字则表示该原子只出现1次。
在一个分子式中，允许一个元素在不同位置出现多次。分子式的长度不超过100个字符

输出描述

输出一个正整数，即计算所得的相对分子量

样例输入

C4H9OH

样例输出

74

思路

• map存储相对原子质量的值
• 切割字符比较key，取出value值累加
• 数字-1与对应value相乘后累加

代码

import java.util.*;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        char[] a = sc.nextLine().toCharArray();
        HashMap<String, Integer> map = getMap();
        int sum = 0;
        String n = "";
        for (int i = 0; i < a.length; i++) {
            String s = a[i] + "";
            if (map.containsKey(s)) {
                sum += map.get(s);
                n = s;
            } else {
                int x = a[i] - 48;
                sum += (x - 1) * map.get(n);
            }
        }
        System.out.println(sum);
    }

    public static HashMap<String, Integer> getMap() {
        HashMap<String, Integer> map = new HashMap<>();
        map.put("C", 12);
        map.put("H", 1);
        map.put("O", 16);
        map.put("N", 14);
        return map;
    }
}

复杂度分析

• 时间复杂度：O(N)
• 空间复杂度：O(1)