본문 바로가기

⏳ 알고리즘/python 알고리즘 문제 풀이

프로그래머스 - LV2. 방문 길이

문제

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

 

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

 

코드

def solution(dirs):
    visited = set()
    x, y = 5, 5
    for d in dirs:
        if d == 'U' and y<10:
            visited.add(((x, y), (x,y+1)))
            y += 1
        if d == 'D' and y>0:
            visited.add(((x, y-1), (x,y)))
            y -= 1
        if d == 'R' and x<10:
            visited.add(((x, y), (x+1,y)))
            x += 1
        if d == 'L' and x>0:
            visited.add(((x-1, y), (x,y)))
            x -= 1
    
    return len(visited)