原创

LeetCode | 机器人能否返回原点

本文字数:

1576

,大约阅读2分钟

放假的时间已经过去一半了,每天坚持看一个多小时的书,时间虽然不多,但是能专心把书看进去就可以了。今天分享的是 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。至于其他的解决方法,就不再去想了。

数据结构
算法
LeetCode
面试
字符串
模拟
  • 作者:Netor0x86(联系作者)
  • 发表时间:2019-10-03 15:01
  • 版权声明:自由转载-非商用-非衍生-保持署名(创意共享3.0许可证)
  • 公众号转载:请在文末添加作者公众号二维码
  • 评论