Home > Baekjoon > 소용돌이 예쁘게 출력하기

소용돌이 예쁘게 출력하기
2 초 128 MB 30.452%

출처 : 소용돌이 예쁘게 출력하기

Solution
Python

import sys
input = sys.stdin.readline

# 1022번 문제를 풀기 위한 핵심 함수 (r=y(열), c=x(행))
def cal_vortex(x, y):
    # x, y 둘 다 0일 때
    if x == 0 and y == 0: return 1
    # 다른 경우 계산 시작
    n = max(abs(x), abs(y))
    m = min(abs(x), abs(y))
    grad = -y/x if x != 0 else 5000
    # 1사분면에서 기울기가 1보다 작을 때(x축에 가까울 때), 1보다 클 때(y축에 가까울 때)
    if (x >= 0 and y <= 0) and grad <= 1: var = 1+m/n
    elif (x >= 0 and y <= 0) and grad >= 1: var = 3-m/n
    # 2사분면에서 기울기가 -1보다 작을 때(y축에 가까울 때), -1보다 클 때(x축에 가까울 때)
    elif (x <= 0 and y <= 0) and grad <= -1: var = 3+m/n
    elif (x <= 0 and y <= 0) and grad >= -1: var = 5-m/n
    # 3사분면에서 기울기가 1보다 작을 때(x축에 가까울 때), 1보다 클 때(y축에 가까울 때)
    elif (x <= 0 and y >= 0) and grad <= 1: var = 5+m/n
    elif (x <= 0 and y >= 0) and grad >= 1: var = 7-m/n
    # 4사분면에서 기울기가 -1보다 작을 때(y축에 가까울 때), -1보다 클 때(x축에 가까울 때)
    elif (x >= 0 and y >= 0) and grad <= -1: var = 7+m/n
    elif (x >= 0 and y >= 0) and grad >= -1: var = 1-m/n

    return int(1+var*n+4*n*(n-1))

r1, c1, r2, c2 = map(int, input().split())

num_r, num_c = r2-r1+1, c2-c1+1
answer = [[0] * num_c for _ in range(num_r)]

high_num = 0
for ay, ry in enumerate(range(r1, r2+1)):
    for ax, cx in enumerate(range(c1, c2+1)):
        answer[ay][ax] = cal_vortex(cx, ry)
        if high_num < answer[ay][ax]: high_num = answer[ay][ax]

for y in range(num_r):
    for x in range(num_c):
        print(f'{answer[y][x]:>{len(str(high_num))}d}', end=' ')
    print()

문제

크기가 무한인 정사각형 모눈종이가 있다. 모눈종이의 각 정사각형은 행과 열의 쌍으로 표현할 수 있다.

이 모눈종이 전체를 양의 정수의 소용돌이 모양으로 채울 것이다. 일단 숫자 1을 0행 0열에 쓴다. 그리고 나서 0행 1열에 숫자 2를 쓴다. 거기서 부터 소용돌이는 반시계 방향으로 시작된다. 다음 숫자는 다음과 같이 채우면 된다.


    -3 -2 -1  0  1  2  3

    --------------------

-3 |37 36 35 34 33 32 31

-2 |38 17 16 15 14 13 30

-1 |39 18  5  4  3 12 29

 0 |40 19  6  1  2 11 28

 1 |41 20  7  8  9 10 27

 2 |42 21 22 23 24 25 26

 3 |43 44 45 46 47 48 49

이 문제는 위와 같이 채운 것을 예쁘게 출력하면 된다. r1, c1, r2, c2가 입력으로 주어진다. r1, c1은 가장 왼쪽 위 칸이고, r2, c2는 가장 오른쪽 아래 칸이다.

예쁘게 출력한다는 것은 다음과 같이 출력하는 것이다.

  1. 출력은 r1행부터 r2행까지 차례대로 출력한다.
  2. 각 원소는 공백으로 구분한다.
  3. 모든 행은 같은 길이를 가져야 한다.
  4. 공백의 길이는 최소로 해야 한다.
  5. 모든 숫자의 길이(앞에 붙는 공백을 포함)는 같아야 한다.
  6. 만약 수의 길이가 가장 길이가 긴 수보다 작다면, 왼쪽에서부터 공백을 삽입해 길이를 맞춘다.

입력

첫째 줄에 네 정수 r1, c1, r2, c2가 주어진다.

출력

r2 - r1 + 1개의 줄에 소용돌이를 예쁘게 출력한다.

제한

  • -5 000 ≤ r1, c1, r2, c2 ≤ 5,000
  • 0 ≤ r2 - r1 ≤ 49
  • 0 ≤ c2 - c1 ≤ 4

예제 입력 1

-3 -3 2 0

예제 출력 1

37 36 35 34
38 17 16 15
39 18  5  4
40 19  6  1
41 20  7  8
42 21 22 23

예제 입력 2

-2 2 0 3

예제 출력 2

13 30
12 29
11 28

예제 입력 3

-1 -2 -1 1

예제 출력 3

18  5  4  3

예제 입력 4

0 0 0 0

예제 출력 4

1