[Python] BOJ 10157번. 자리배정

작성일 | In |

10157번. 자리배정

문제 링크

풀이 코드

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
# 10157번. 자리배정


import sys
input = sys.stdin.readline

c, r = map(int, input().split())
k = int(input())
box = [[0 for i in range(c)] for i in range(r)]

if k > r*c:  # 좌석을 배치할 수 없을 때
    print(0)
    sys.exit()

cnt = 1  # 배정될 좌석
dir = 0  # 방향
dx = [-1, 0, 1, 0]  # 상우하좌
dy = [0, 1, 0, -1]

cr = r-1
cc = 0
while True:
    if cnt == k:
        print((cc+1), (r-cr))
        break
    box[cr][cc] = cnt
    nr = cr+dx[dir]
    nc = cc+dy[dir]
    if nr < 0 or nc < 0 or nr >= r or nc >= c or box[nr][nc] != 0:
        dir = (dir+1) % 4  # 방향전환
        nr = cr + dx[dir]
        nc = cc + dy[dir]

    cr = nr
    cc = nc
    cnt += 1

비고