[python 파이썬][프로그래머스] 방문길이

 출처 :  https://programmers.co.kr/learn/courses/30/lessons/49994

코딩테스트 연습 - 방문 길이

게임 캐릭터를 4가지 명령어를 통해 움직이려 합니다. 명령어는 다음과 같습니다.

 

- U: 위쪽으로 한 칸 가기

- D: 아래쪽으로 한 칸 가기

- R: 오른쪽으로 한 칸 가기

- L: 왼쪽으로 한 칸 가기

 

캐릭터는 좌표평면의 (0, 0) 위치에서 시작합니다. 좌표평면의 경계는 왼쪽 위(-5, 5), 왼쪽 아래(-5, -5), 오른쪽 위(5, 5), 오른쪽 아래(5, -5)로 이루어져 있습니다.

예를 들어, ULURRDLLU로 명령했다면

• 1번 명령어부터 7번 명령어까지 다음과 같이 움직입니다.

• 8번 명령어부터 9번 명령어까지 다음과 같이 움직입니다.

이때, 우리는 게임 캐릭터가 지나간 길 중 캐릭터가 처음 걸어본 길의 길이를 구하려고 합니다. 예를 들어 위의 예시에서 게임 캐릭터가 움직인 길이는 9이지만, 캐릭터가 처음 걸어본 길의 길이는 7이 됩니다. (8, 9번 명령어에서 움직인 길은 2, 3번 명령어에서 이미 거쳐 간 길입니다)

단, 좌표평면의 경계를 넘어가는 명령어는 무시합니다.

예를 들어, LULLLLLLU로 명령했다면

• 1번 명령어부터 6번 명령어대로 움직인 후, 7, 8번 명령어는 무시합니다. 다시 9번 명령어대로 움직입니다.

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

제한사항

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

입출력 예

dirs                                                                          answer

'ULURRDLLU'	7
'LULLLLLLU'	7

 

---

◆ 첫번째로 시도한 풀이(실패)

- 연속으로 왔다갔다 하는것(ex. (0,0) -> (0,1) -> (0,0)을 다른 문자로 치환하여 처리

- 지나갔던 곳으로 다시 온 경우, 중복이 딱 한번이라면 끝맺음이거나 교차하는 경우로 따로 처리

def solution(dirs):
    start = [0,0]
    stack = []
    dupli = []
    last = []
    answer = 0
	# 연속으로 왔다갔다 하는 것 제거
    dirs = dirs.replace('UD','a')
    dirs = dirs.replace('DU','a')
    dirs = dirs.replace('LR','b')
    dirs = dirs.replace('RL','b')  
    for i in range(len(dirs)):
	# 연속으로 왔다갔다 하는 것 처리
        if dirs[i] == 'a' or dirs[i] == 'b':
            dupli.append(dirs[i])
            continue
        if dirs[i] == 'U':
            start = [start[0],start[1] + 1]
        elif dirs[i] == 'D':
            start = [start[0],start[1] - 1]
        elif dirs[i] == 'R':
            start = [start[0] + 1,start[1]]
        elif dirs[i] == 'L':
            start = [start[0] - 1,start[1]]
		#범위 밖으로 벗어났을 경우
        if abs(start[0]) > 5 or abs(start[1]) > 5:
            start = stack[-1]
            continue
        if start not in stack:
            stack.append(start)
        else :
		# 지나갔던 곳으로 다시 온 경우, 중복이 딱 한번이라면 끝맺음이거나 교차하는 경우임
            if last == [] or last[-1] not in stack:
                last.append(start)
    return len(stack) + len(set(dupli)) +len(last)

• 위 문제가 실패한 이유는 길이의 개념으로이 아닌 점의 개념으로 처리했기 때문이다.
• 여기서 중요한 것은 길이의 개념과 양방향의 개념이다.
• 길이의 개념 : 점이 이동할때 이동하기 전 위치도 기록을 해야 한다는 것이다.
• 양방향의 개념 : 왔던 곳을 다시가는 것은 처음 본 길이 아니기 때문에 제외
• 1. (0,0), (1,0), (0,0) 이런식으로 이동한다고 할 때 길이만 기록하게되면 (0,0,1,0) (1,0,0,0) 이렇게 된다.
• 2. 따라서 (0,0,1,0)→(x,y,nx,ny)뿐만 아니라 왔던곳과 이동한 곳을 바꿔준(1,0,0,0)→(nx,ny,x,y)도 기록해야 한다.

• 또한 stack을 set구조로 하여 중복을 제거하면 보다 손쉽게 문제를 해결할 수 있다.

◆ 두번째로 시도한 풀이(성공)

def solution(dirs):
    x,y,nx,ny = 0,0,0,0
    stack = set()
    answer = 0
    for i in range(len(dirs)):
        if dirs[i] == 'U':
            ny += 1
        elif dirs[i] == 'D':
            ny -= 1
        elif dirs[i] == 'R':
            nx += 1
        elif dirs[i] == 'L':
            nx -= 1
        if abs(nx) > 5 or abs(ny) > 5:
            nx, ny = x, y
            continue
        if (x,y,nx,ny) not in stack:
            stack.add((x,y,nx,ny))
            stack.add((nx,ny,x,y))
            answer += 1
        x,y = nx, ny
        print(stack)
    return answer

 

◆ 다른사람의 풀이(성공)

    - ULDR이동을 dict방식으로 하여 구현

def solution(dirs): 
    dxs, dys = [-1, 0, 1, 0], [0, -1, 0, 1] 
    d = {"U": 0, "L":1, "D":2, "R": 3} 
    visited = set() 
    answer = 0 
    x, y = 0, 0 
    for dir in dirs: 
        i = d[dir] 
        nx, ny = x + dxs[i], y + dys[i] 
        if nx < -5 or nx > 5 or ny < -5 or ny > 5: continue 
        if (x, y, nx, ny) not in visited: 
            visited.add((x, y, nx, ny)) 
            visited.add((nx, ny, x, y)) # 길은 '양방향'
            answer += 1
            print((x, y, nx, ny),(nx, ny, x, y))
        x, y = nx, ny 
        
    return answer