leetcode657：机器人能否返回原点（robot-return-to-origin）

[carloscn]

问题描述

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

移动顺序由字符串 moves 表示。字符 move[i] 表示其第 i 次移动。机器人的有效动作有 R（右），L（左），U（上）和 D（下）。

如果机器人在完成所有动作后返回原点，则返回 true。否则，返回 false。

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

 

示例 1:

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

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

提示:

1 <= moves.length <= 2 * 104 moves 只包含字符 'U', 'D', 'L' 和 'R'

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

[carloscn]

问题分析

C 语言版

static int32_t is_move_back(const char* vector, bool *out)
{
    int32_t ret = 0;
    size_t i = 0, len = 0;
    int32_t lr = 0;
    int32_t ud = 0;

    UTILS_CHECK_PTR(vector);
    UTILS_CHECK_PTR(out);
    UTILS_CHECK_LEN(len = strlen(vector));

    for (i = 0; i < len; i ++) {
        switch (vector[i]) {
            case 'L':
            case 'l':
            lr ++;
            break;

            case 'R':
            case 'r':
            lr --;
            break;

            case 'U':
            case 'u':
            ud ++;
            break;

            case 'D':
            case 'd':
            ud --;
            break;

            default:
            *out = false;
            goto finish;
            break;
        }
    }

    if (ud == 0 && lr == 0) {
        *out = true;
    } else {
        *out = false;
    }

finish:
    return ret;
}

rust版本

fn is_move_back(vector: &str) -> bool
{
    let mut pos = (0_i32 ,0_i32);

    if vector.len() == 0 {
        return true;
    }

    for e in vector.chars() {
        match e {
            'l' | 'L' => pos.0 += 1,
            'r' | 'R' => pos.0 -= 1,
            'u' | 'U' => pos.1 += 1,
            'd' | 'D' => pos.1 -= 1,
            _ => return false
        }
    }

    if pos == (0 ,0) {
        return true;
    } else {
        return false;
    }
}

[carloscn]

code:

https://github.com/carloscn/structstudy/blob/master/c_programming/str/36_robot-return-to-origin_657.c https://github.com/carloscn/structstudy/blob/master/rust_programming/str/src/n36_robot_return_to_origin_657.rs

result: