leetcode 速算机器人

小扣在秋日市集发现了一款速算机器人。店家对机器人说出两个数字（记作 x 和 y），请小扣说出计算指令：

"A" 运算：使 x = 2 * x + y； "B" 运算：使 y = 2 * y + x。 在本次游戏中，店家说出的数字为 x = 1 和 y = 0，小扣说出的计算指令记作仅由大写字母 A、B 组成的字符串 s，字符串中字符的顺序表示计算顺序，请返回最终 x 与 y 的和为多少。

示例 1：

输入：s = "AB"

输出：4

解释：
经过一次 A 运算后，x = 2, y = 0。
再经过一次 B 运算，x = 2, y = 2。
最终 x 与 y 之和为 4。

提示：

• 0 <= s.length <= 10
• s 由 'A' 和 'B' 组成

我的算法实现：

/**
 * @param {string} s
 * @return {number}
 */
var calculate = function (s) {
  // 保存计算规则
  const calcObj = {
    A: (x, y) => {
      return [2 * x + y, y]
    },
    B: (x, y) => {
      return [2 * y + x, x]
    }
  }
  let x = 1, y = 0;
  for (let i = 0; i < s.length; i ++) {
  	// 直接将结果保存起来，这样就不用判断了
    [x, y] = calcObj[s.charAt(i)](x, y)
  }
  return x + y;
};

这个方法的代码也有不少，同时效率也很一般，总共用时 84ms ，而直接写的耗时为 80ms 。

/**
 * @param {string} s
 * @return {number}
 */
var calculate = function (s) {
  let x = 1,
    y = 0;
  for (let i = 0; i < s.length; i++) {
    if (s.charAt(i) === 'A') {
      x = 2 * x + y;
    } else {
      y = 2 * y + x;
    }
  }
  return x + y;
};

这个就是普通的方法，但是代码量什么的都比前面的少。这种易读性更好。

对应的 java 代码为：

class Solution {
    public int calculate(String s) {
        int x = 1, y = 0;
        for (int i = 0; i < s.length(); i++) {
            if (s.charAt(i) == 'A') {
                x = 2 * x + y;
            } else {
                y = 2 * y + x;
            }
        }
        return x + y;
    }
}

不知道为啥，我用 java 实现总是显示 0ms。

有一个大神直接使用数学方法解决了这个问题，非常妙。

class Solution {
  public int calculate(String s) {
    return (int)Math.pow(2, s.length());
  }
}

对应的 javascript 代码为：

var calculate = function (s) {
  return 2 ** s.length;
};

只不过这个执行耗时居然比前面的还要多，高达 100ms 。

直接看他的解释： 不需要遍历，有数学方法

来源：力扣（LeetCode）