[programmers] 방문 길이 - Summer/Winter Coding(~2018)

코딩 테스트 연습 - Summer/Winter Coding(~2018)

---

<방문 길이>

문제 설명

 

게임 캐릭터를 4가지 명령어를 통해 움직이려 합니다. 명령어는 다음과 같습니다.
U: 위쪽으로 한 칸 가기
D: 아래쪽으로 한 칸 가기
R: 오른쪽으로 한 칸 가기
L: 왼쪽으로 한 칸 가기
캐릭터는 좌표평면의 (0, 0) 위치에서 시작합니다.
좌표평면의 경계는 왼쪽 위(-5, 5), 왼쪽 아래(-5, -5), 오른쪽 위(5, 5), 오른쪽 아래(5, -5)로 이루어져 있습니다.

이때, 우리는 게임 캐릭터가 지나간 길 중 캐릭터가 처음 걸어본 길의 길이를 구하려고 합니다.

명령어가 매개변수 dirs로 주어질 때, 게임 캐릭터가 처음 걸어본 길의 길이를 구하여
return 하는 solution 함수를 완성해 주세요.

 

제한 사항

 

- dirs는 string형으로 주어지며, 'U', 'D', 'R', 'L' 이외에 문자는 주어지지 않습니다.
- dirs의 길이는 500 이하의 자연수입니다.

 

 

문제 풀이

 

 처음에 내가 작성한 코드는 아래와 같다 

 

 - 틀린 코드

def solution(dirs):
    answer = 1

    x, y = 0, 0
    arr = []
    for i in dirs:
        check=False
        if i == 'U':
            if y < 5:
                y += 1
            else:
                check=True
        elif i == 'D':
            if y > -5:
                y -= 1
            else:
                chck=True
        elif i == 'R':
            if x < 5:
                x += 1
            else:
                check=True
        elif i == 'L':
            if x > -5:
                x -= 1
            else:
                check=True
        if check!=True:
            arr.append([x, y])
    print(arr)

    check=False
    for i in range(1, len(arr)):
        temp=arr[:i]
        cnt=0
        for j in range(len(temp)):
            if arr[i][0]==temp[j][0] and arr[i][1]==temp[j][1]:
                check=True
            else:
                cnt+=1
        if cnt==len(temp):
            answer+=1
    if check==True:
        answer+=1

    return answer

 

예시로 주어진 2문제는 맞는 것 같았지만 채점 결과에서 후두둑😥

뭐가 잘못된 코드인가 살펴보니 이 방법으로는 해결할 수 없을 것 같다는 생각이 들었다.

결국 검색의 힘을 받았다...😢

 

set을 사용하여 해결하는 것이었다. 

왜 이걸 생각해내지 못했는지 충격적이었다.

 

최종 작성한 나의 코드이다.

 - my solution

def solution(dirs):
    answer = 0
    
    x,y=0,0
    arr = set()

    for i in dirs: # U,D,R,L
        start_x, start_y = x, y #시작 위치 update
        if i == 'U' and y < 5:
            y += 1
        elif i == 'D' and y > -5:
            y -= 1
        elif i == 'R' and x < 5:
            x += 1
        elif i == 'L' and x > -5:
            x -= 1

        if start_x==x and start_y==y: # 좌표를 벗어나 같은 자리에 머물면 
            pass
        else: #양방향으로 넣어줌
            arr.add((start_x, start_y, x, y))
            arr.add((x, y, start_x, start_y))
        
    answer=len(arr)//2 #양방향으로 저장했으므로 나누기 2
    
    return answer

 

이 코드로 통과했지만 이것도 잘 짠 코드가 아니라는 생각이 들었다.

다른 코드를 본 결과 U, D, R, L을 처리하는 부분에서 list나 dict를 활용하여 더 짧게 

구현할 수 있었다.

 

1) set을 선언: 중복된 길을 제거하기 위해서이다. (처음 걸어본 길만)

 

2) U, D, R, L을 처리하는 반복문을 작성

① 문제에서 좌표 평면의 경계가 (-5, 5), (-5, -5), (5, 5), (5, -5)라고 주어졌으므로 조건을 함께 작성

② 좌표 이동에서 경계 조건을 걸었으므로 시작 위치와 이동 위치가 같은 경우는 set에 추가 x

③ set에 (시작 좌표 x, 시작 좌표 y, 이동 후 x, 이동 후 y), (이동 후 x, 이동 후 y, 시작 좌표 x, 시작 좌표 y)

를 양방향으로 넣어준다. (처음 걸어본 길만 count 하기 위해)

 

3) 양방향으로 set에 저장했으므로 처음 가본 길의 길이는 set을 2로 나눈 몫

---

생각🤔

 

set을 활용하여 양방향으로 저장해준다는 것을 보자마자 '아차' 했다.

다음번에 이러한 문제가 나왔을 때 탁 알아차리고 풀 수 있으면 좋겠다.

 

내가 생각하기에 이 문제의 포인트

1) set 활용

2) 양방향 저장

---

출처: 프로그래머스 코딩 테스트 연습, https://programmers.co.kr/learn/challenges