구현 씹어먹기

게시 2023/07/08

By 정민 12 분읽는 시간

구현 알고리즘은,

머릿속에 있는 알고리즘을 소스코드로 바꾸는 과정
풀이를 떠올리는 것은 쉽지만 소스코드로 옮기기 어려운 문제
완전 탐색
- 모든 경우의 수를 다 계산하는 해결 방법
시뮬레이션
- 문제에서 제시한 알고리즘을 한 단계씩 차례대로 직접 수행
구현 시 고려해야 할 메모리 제약 사항
- 리스트 크기 제약
  - 대체로 코딩 테스트에서는 128~512MB로 메모리를 제한
  - 리스트를 여러 개 선언하고, 그중 크기가 1,000만 이상인 리스트가 있다면 메모리를 고려할 것

예제 1. 상하좌우

여행가 A는 N $\times$ N 크기의 정사각형 공간 위에 서 있다. 이 공간은 1 $\times$ 1 크기의 정사각형으로 나누어져 있다. 가장 왼쪽 위 좌표는 (1, 1)이며, 가장 오른쪽 아래 좌표는 (N, N)에 해당한다. 여행가 A는 상, 하, 좌, 우 방향으로 이동할 수 있으며, 시작 좌표는 항상 (1, 1)이다. 우리 앞에는 여행가 A가 이동할 계획이 적힌 계획서가 놓여 있다.

계획서에는 하나의 줄에 띄어쓰기를 기준으로 하여 L, R, U, D 중 하나의 문자가 반복적으로 적혀 있다. 각 문자의 의미는 다음과 같다.

L: 왼쪽으로 한 칸 이동
R: 오른쪽으로 한 칸 이동
U: 위로 한 칸 이동
D: 아래로 한 칸 이동

이때 여행가 A가 N $\times$ N 크기의 정사각형 공간을 벗어나는 움직임은 무시된다. 예를 들어 (1, 1)의 위치에서 L 또는 U를 만나면 무시된다. 계획서가 주어졌을 때 여행가 A가 최종적으로 도착할 지점의 좌표를 출력하는 프로그램을 작성하시오.

입력 조건
- 첫째 줄에 공간의 크기를 나타내는 N이 주어진다. ($1\leq N\leq 100$)
- 둘째 줄에 여행가 A가 이동할 계획서 내용이 주어진다. ($1\leq 이동 횟수 \leq 100$)
출력 조건
- 첫째 줄에 여행가 A가 최종적으로 도착할 지점의 좌표 (X, Y)를 공백으로 구분하여 출력한다.

  
n = int(input())
plans = map(str, input().split())
here = [1,1]

for plan in plans:
    if plan == 'L':
        if here[1] != 1:
            here[1] -= 1
    elif plan == 'R':
        if here[1] != n:
            here[1] += 1
    elif plan == 'U':
        if here[0] != 1:
            here[0] -= 1
    elif plan == 'D':
        if here[0] != n:
            here[0] += 1

print(f'({here[0],here[1]})')

예제 2. 시각

정수 N이 입력되면 00시 00분 00초부터 N시 59분 59초까지의 모든 시각 중에서 3이 하나라도 포함되는 모든 경우의 수를 구하는 프로그램을 작성하시오.

입력 조건
- 첫째 줄에 정수 N이 입력된다. ($0 \leq N \leq 23$)
출력 조건
- 00시 00분 00초부터 N시 59분 59초까지의 모든 시각 중에서 3이 하나라도 포함되는 모든 경우의 수를 출력한다.

  
n = int(input())
count = 0

for i in range(n+1):
    for j in range(60):
        for k in range(60):
            if '3' in str(i)+str(j)+str(k):
                count += 1

print(count)

실전 1. 왕실의 나이트

행복 왕국의 왕실 정원은 체스판과 같은 $8\times8$ 좌표 평면이다. 왕실 정원의 특정한 한 칸에 나이트가 서 있다. 나이트는 매우 충성스러운 신하로서 매일 무술을 연마한다.

나이트는 말을 타고 있기 때문에 이동을 할 때는 L자 형태로만 이동할 수 있으며 정원 밖으로는 나갈 수 없다. 나이트는 특정한 위치에서 다음과 같은 2가지 경우로 이동할 수 있다.

수평으로 두 칸 이동한 뒤에 수직으로 한 칸 이동하기
수직으로 두 칸 이동한 뒤에 수평으로 한 칸 이동하기

$8\times8$ 좌표 평면상에서 나이트의 위치가 주어졌을 때 나이트가 이동할 수 있는 경우의 수를 출력하는 프로그램을 작성하시오. 이때 왕실의 정원에서 행 위치를 표현할 때는 1부터 8로 표현하며, 열 위치를 표현할 때는 a부터 h로 표현한다.

입력 조건
- 첫째 줄에 $8\times8$ 좌표 평면상에서 현재 나이트가 위치한 곳의 좌표를 나타내는 두 문자로 구성된 문자열이 입력된다. 입력 문자는 a1처럼 열과 행으로 이뤄진다.
출력 조건
- 첫째 줄에 나이트가 이동할 수 있는 경우의 수를 출력하시오.

  
here = input()
column = ord(here[0])-96
row = int(here[1])
count = 0
steps = [(2, -1), (2, 1), (-2, -1), (-2, 1), (1, 2), (1, -2), (-1, 2), (-1, -2)]

for step in steps:
    new_row = row + step[0]
    new_col = column + step[1]
    if new_row >= 1 and new_row <= 8 and new_col >= 1 and new_col <= 8:
        count += 1

print(count)

실전2. 게임 개발

현민이는 게임 캐릭터가 맵 안에서 움직이는 시스템을 개발 중이다. 캐릭터가 있는 장소는 $1\times 1$ 크기의 정사각형으로 이뤄진 $N\times M$ 크기의 직사각형으로, 각각의 칸은 육지 또는 바다이다. 캐릭터는 동서남북 중 한 곳을 바라본다.

맵의 각 칸은 (A, B)로 나타낼 수 있고, A는 북쪽으로부터 떨어진 칸의 개수, B는 서쪽으로부터 떨어진 칸의 개수이다. 캐릭터는 상하좌우로 움직일 수 있고, 바다로 되어 있는 공간에는 갈 수 없다. 캐릭터의 움직임을 설정하기 위해 정해 놓은 매뉴얼은 이러하다.

현재 위치에서 현재 방향을 기준으로 왼쪽 방향(반시계 방향으로 90도 회전한 방향)부터 차례대로 갈 곳을 정한다.
캐릭터의 바로 왼쪽 방향에 아직 가보지 않은 칸이 없다면, 왼쪽 방향으로 회전한 다음 왼쪽으로 한 칸을 전진한다. 왼쪽 방향에 가보지 않은 칸이 없다면, 왼쪽 방향으로 회전만 수행하고 1단계로 돌아간다.
만약 네 방향 모두 이미 가본 칸이거나 바다로 되어 있는 칸인 경우에는, 바라보는 방향을 유지한 채로 한 칸 뒤로 가고 1단계로 돌아간다. 단, 이때 뒤쪽 방향이 바다인 칸이라 뒤로 갈 수 없는 경우에는 움직임을 멈춘다.

현민이는 위 과정을 반복적으로 수행하면서 캐릭터의 움직임에 이상이 있는지 테스트하려고 한다. 매뉴얼에 따라 캐릭터를 이동시킨 뒤에, 캐릭터가 방문한 칸의 수를 출력하는 프로그램을 만드시오.

입력 조건
- 첫째 줄에 맵의 세로 크기 N과 가로 크기 M을 공백으로 구분하여 입력한다. ($3\leq N,M \leq 50$)
- 둘째 줄에 게임 캐릭터가 있는 칸의 좌표 (A, B)와 바라보는 방향 d가 각각 서로 공백으로 구분하여 주어진다. 방향 d의 값으로는 다음과 같이 4가지가 존재한다.
  - 0: 북쪽
  - 1: 동쪽
  - 2: 남쪽
  - 3: 서쪽
- 셋째 줄부터 맵이 육지인지 바다인지에 대한 정보가 주어진다. N개의 줄에 맵의 상태가 북쪽부터 남쪽 순서대로, 각 줄의 데이터는 서쪽부터 동쪽 순서대로 주어진다. 맵의 외곽은 항상 바다로 되어 있다.
  - 0: 육지
  - 1: 바다
- 처음에 게임 캐릭터가 위치한 칸의 상태는 항상 육지이다.
출력 조건
- 첫째 줄에 이동을 마친 후 캐릭터가 방문한 칸의 수를 출력한다.

  
n, m = map(int, input().split())
a, b, d = map(int, input().split())
sea = []
step = [(-1,0),(0,1),(1,0),(0,-1)]

for i in range(n):
    sea.append(list(map(int, input().split())))

here = [[0]*m for _ in range(n)]
here[a][b] = 1
count = 0
turn = 0

def turn_left(d):
    d -= 1

    if d == -1:
        d = 3

    return d

while True:
    d = turn_left(d)
    na = a+step[d][0]
    nb = b+step[d][1]

    if here[na][nb] == 0 and sea[na][nb] == 0:
        here[na][nb] = 1
        a = na
        b = nb
        count += 1
        turn = 0
        continue
    else:
        turn += 1
    
    if turn == 4:
        na = a - step[d][0]
        nb = b - step[d][1]
        if sea[na][nb] == 0:
            a = na
            b = nb
            count += 1
        else:
            break
        turn = 0


print(count)

Algorithm