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

<방문 길이>

문제(출처: https://school.programmers.co.kr/learn/courses/30/lessons/49994)

 

 

문제 풀이

 

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

 

 - 틀린 코드

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