Leetcode657. 机器人能否返回原点

题目

在二维平面上，有一个机器人从原点 (0, 0) 开始。给出它的移动顺序，判断这个机器人在完成移动后是否在 (0, 0) 处结束。

移动顺序由字符串表示。字符 move[i] 表示其第 i 次移动。机器人的有效动作有 R（右），L（左），U（上）和 D（下）。如果机器人在完成所有动作后返回原点，则返回 true。否则，返回 false。

注意：机器人“面朝”的方向无关紧要。 “R” 将始终使机器人向右移动一次，“L” 将始终向左移动等。此外，假设每次移动机器人的移动幅度相同。

示例 1:

输入: "UD"
输出: true
解释：机器人向上移动一次，然后向下移动一次。所有动作都具有相同的幅度，因此它最终回到它开始的原点。因此，我们返回 true。

示例 2:

输入: "LL"
输出: false
解释：机器人向左移动两次。它最终位于原点的左侧，距原点有两次 “移动” 的距离。我们返回 false，因为它在移动结束时没有返回原点。

思路

统计各个字母出现的次数，比较字母U和D、L和R的出现次数是否相等。为了方便用数组统计，大写字母'A'到'Z'和0b111111进行与运算的值为1-26, 由于这里最大字符是'U', 取数组的长度为22即可。

C++代码

#include <iostream>
#include <vector>
#include <map>
#include <set>
using namespace std;
class Solution {
public:
    bool judgeCircle(string moves) {
        vector<int> count(22, 0);
        const int value = 0b111111;
        for (char c: moves) ++count[c & value];
        return count['U' & value] == count['D' & value] && count['L' & value] == count['R' & value];
    }
};
int main(int argc, const char * argv[]) {
    // insert code here...
    Solution solution;
    
    cout << solution.judgeCircle("UUDDLLRR") << endl;
    cout << solution.judgeCircle("UDUDLLLRLR") << endl;
    return 0;
}

换成c数组统计，还可以更快:

class Solution {
public:
    bool judgeCircle(string moves) {
        uint16_t count[13] = {0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0};
        const int value = 0b1111;
        for (char c: moves) ++count[c & value];
        return count['U' & value] == count['D' & value] && count['L' & value] == count['R' & value];
    }
};

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/robot-return-to-origin