[프로그래머스] 산책 (lv.0)

 

return = [동쪽으로 떨어진 거리, 북쪽으로 떨어진 거리]

 

 

 

자바

class Solution{
	public int[] solution(String route){
    	int east = 0;
        int north = 0;
        int[] answer = new int [2];
        for(int i=0; i<route.length(); i++){
            switch(route.charAt(i)){
                case 'N':
                    north++;
                    break;
                case 'S':
                    north--;
                    break;
                case 'E':
                    east++;
                    break;
                case 'W':
                    east--;
                    break;
                }
            }
        answer[0] = east;
        answer[1] = north;
        return answer;
    }
}

 

 

 

 

파이썬

def solution(route):
    east = 0
    north = 0
    for i in route:
        if i == "N":
            north += 1
        elif i == "S":
            north -= 1
        elif i == "E":
            east += 1
        elif i == "W":
            east -= 1
    return [east, north]