[코테 풀이] Robot Return to Origin

시내·2024년 6월 7일

Q_657) Robot Return to Origin

출처 : https://leetcode.com/problems/robot-return-to-origin/?envType=study-plan-v2&envId=programming-skills

There is a robot starting at the position (0, 0), the origin, on a 2D plane. Given a sequence of its moves, judge if this robot ends up at (0, 0) after it completes its moves.

You are given a string moves that represents the move sequence of the robot where moves[i] represents its ith move. Valid moves are 'R' (right), 'L' (left), 'U' (up), and 'D' (down).

Return true if the robot returns to the origin after it finishes all of its moves, or false otherwise.

Note: The way that the robot is "facing" is irrelevant. 'R' will always make the robot move to the right once, 'L' will always make it move left, etc.
Also, assume that the magnitude of the robot's movement is the same for each move.

class Solution {
    public boolean judgeCircle(String moves) {
        int[] counts = new int[2]; //in the order of : UD && RL
        for (int i = 0; i < moves.length(); i++) {
            if (moves.charAt(i) == 'U') {
                counts[0]++;
            } else if (moves.charAt(i) == 'D') {
                counts[0]--;
            } else if (moves.charAt(i) == 'R') {
                counts[1]++;
            } else if (moves.charAt(i) == 'L') {
                counts[1]--;
            }
        }
        if (counts[0] == 0 && counts[1] == 0) {
            return true;
        }
        return false;
    }
}
profile
contact 📨 ksw08215@gmail.com

0개의 댓글