Development/Algorithm

[이코테 2021] 4장 구현 [예제 4-1] 상하좌우

jstar0525 2022. 5. 3. 07:01
반응형

나의 풀이

n = int(input())
cmd = input().split()

x, y = 1, 1

move_types = ['L', 'R', 'U', 'D']

dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]

for c in cmd:
    for i in range(len(move_types)):
        if c == move_types[i]:
            tmp_x = x + dx[i]
            tmp_y = y + dy[i]
    if tmp_x < 1 or tmp_x > n or tmp_y < 1 or tmp_y > n:
        continue
    else:
        x, y = tmp_x, tmp_y

print(x, y)

답안 예시

# N 입력받기
n = int(input())
x, y = 1, 1
plans = input().split()

# L, R, U, D에 따른 이동 방향
dx = [0, 0, -1, 1]
dy = [-1, 1, 0, 0]
move_types = ['L', 'R', 'U', 'D']

# 이동 계획을 하나씩 확인
for plan in plans:
    # 이동 후 좌표 구하기
    for i in range(len(move_types)):
        if plan == move_types[i]:
            nx = x + dx[i]
            ny = y + dy[i]
    # 공간을 벗어나는 경우 무시
    if nx < 1 or ny < 1 or nx > n or ny > n:
        continue
    # 이동 수행
    x, y = nx, ny

print(x, y)
반응형