1. 程式人生 > >[Leetcode] 657. 機器人能否返回原點 java

[Leetcode] 657. 機器人能否返回原點 java

在二維平面上,有一個機器人從原點 (0, 0) 開始。給出它的移動順序,判斷這個機器人在完成移動後是否在 (0, 0) 處結束

移動順序由字串表示。字元 move[i] 表示其第 i 次移動。機器人的有效動作有 R(右),L(左),U(上)和 D(下)。如果機器人在完成所有動作後返回原點,則返回 true。否則,返回 false。

注意:機器人“面朝”的方向無關緊要。 “R” 將始終使機器人向右移動一次,“L” 將始終向左移動等。此外,假設每次移動機器人的移動幅度相同。

示例 1:

輸入: "UD"
輸出: true
解釋:機器人向上移動一次,然後向下移動一次。所有動作都具有相同的幅度,因此它最終回到它開始的原點。因此,我們返回 true。

示例 2:

輸入: "LL"
輸出: false
解釋:機器人向左移動兩次。它最終位於原點的左側,距原點有兩次 “移動” 的距離。我們返回 false,因為它在移動結束時沒有返回原點。

 開始寫的是用雜湊表,但是測試的時候如果機器人走了很多很多步,這個程式就出錯了。我自己設定了幾個簡單的,比如五六七八步都是沒有問題的。我猜是if((map.get('U')==map.get('D'))&&(map.get('L')==map.get('R')))的問題,運算量太大。

class Solution {
    public boolean judgeCircle(String moves) {
        char[] chars=moves.toCharArray();
        HashMap<Character,Integer> map=new HashMap<Character,Integer>();
        for(int i=0;i<moves.length();i++){
            if(map.containsKey(chars[i])){
                int num=map.get(chars[i])+1;
                map.put(chars[i],num);
            }
            else{
                map.put(chars[i],1);
            }
        }
        if((map.get('U')==map.get('D'))&&(map.get('L')==map.get('R'))){
            return true;
        }
        else{
            return false;
        }
    }
}

後來用了最原始的辦法

class Solution {
    public boolean judgeCircle(String moves) {
        char[] chars=moves.toCharArray();
        int x=0,y=0;
        for(int i=0;i<moves.length();i++){
            if(chars[i]=='U') y++;
            if(chars[i]=='D') y--;
            if(chars[i]=='L') x--;
            if(chars[i]=='R') x++;
        }
        return x==0&&y==0;
    }
}