LeetCode | 机器人能否返回原点

今天分享的是 LeetCode 上面的第 657 题，题目是《机器人能否返回原点》，这也是一道简单的题。

LeetCode 题库的第 657 题——机器人能否返回原点

机器人能否返回原点

题的解法也很简单，先定义坐标，并设置坐标为(0, 0)，然后按照给定的方向去移动，在移动的过程中修改方向，移动完以后再次判断坐标是否为(0, 0)即可。代码如下：

bool judgeCircle(char* moves) {
    int x = 0;
    int y = 0;
    
    int len = strlen(moves);
    
    int i;
    
    for ( i = 0; i < len; i ++ ) {
        switch ( moves[i] ) {
            case 'U': {
                y ++;
                break;
            }
            case 'D': {
                y --;
                break;
            }
            case 'L': {
                x --;
                break;
            }
            case 'R': {
                x ++;
                break;
            }
        }
    }
    
    if ( x == 0 && y == 0 ) {
        return 1;
    }
    
    return 0;
}

LeetCode 有关于代码执行时间和内存消耗的提示，上面的代码执行的时间是 8 ms，可以试着进行优化。试想一下，如果步数是单步，肯定无法回到原点，而步数是偶数步则可能会回到原点，因此可以增加一个判断，增加后的代码如下：

bool judgeCircle(char* moves) {
    int x = 0;
    int y = 0;
    
    int len = strlen(moves);
    
    if ( len % 2 != 0 ) {
        return 0;
    }
    
    int i;
    
    // 这里和上面的代码相同
    
    if ( x == 0 && y == 0 ) {
        return 1;
    }
    
    return 0;
}

增加了上面的判断之后，代码由原来的 8 ms 变成了 4 ms，这也算是有了很大的提高。我试着把 y 变量去掉，只使用一个 x 变量去解决该问题，但是代码会有 bug，虽然 LeetCode 中的测试用例没有测试出来，但是时间又回到了 8 ms。至于其他的解决方法，就不再去想了。

---

我的微信公众号：“码农UP2U”
我的公众号