小扣在秋日市集发现了一款速算机器人。店家对机器人说出两个数字(记作 x 和 y),请小扣说出计算指令:
“A” 运算:使 x = 2 * x + y;
“B” 运算:使 y = 2 * y + x。
在本次游戏中,店家说出的数字为 x = 1 和 y = 0,小扣说出的计算指令记作仅由大写字母 A、B 组成的字符串 s,字符串中字符的顺序表示计算顺序,请返回最终 x 与 y 的和为多少。
0 <= s.length <= 10
s 由 ‘A’ 和 ‘B’ 组成
法一:暴力法:
class Solution {
public:
int calculate(string s) {
int x = 1, y = 0;
for (char c : s) {
if (c == 'A') {
x = 2 * x + y;
} else {
y = 2 * y + x;
}
}
return x + y;
}
};
时间复杂度为O(n),空间复杂度为O(1),n为字符串s的长度。
法二:每次进行A或B运算,其和都会乘2:
class Solution {
public:
int calculate(string s) {
return 1 << s.size(); // string的size函数直接返回string类的长度成员,时间复杂度为O(1)
}
};
时间复杂度为O(1),空间复杂度为O(1)。