728x90
반응형

문제:

크기가 N×M 크기인 배열 A가 있을때, 배열 A의 값은 각 행에 있는 모든 수의 합 중 최솟값을 의미한다. 배열 A가 아래와 같은 경우 1행의 합은 6, 2행의 합은 4, 3행의 합은 15이다. 따라서, 배열 A의 값은 4이다.

 

배열은 회전 연산을 수행할 수 있다. 회전 연산은 세 정수 (r, c, s)로 이루어져 있고, 가장 왼쪽 윗 칸이 (r-s, c-s), 가장 오른쪽 아랫 칸이 (r+s, c+s)인 정사각형을 시계 방향으로 한 칸씩 돌린다는 의미이다. 배열의 칸 (r, c)는 r행 c열을 의미한다.

예를 들어, 배열 A의 크기가 6×6이고, 회전 연산이 (3, 4, 2)인 경우에는 아래 그림과 같이 회전하게 된다.

회전 연산이 두 개 이상이면, 연산을 수행한 순서에 따라 최종 배열이 다르다.

다음은 배열 A의 크기가 5×6이고, 회전 연산이 (3, 4, 2), (4, 2, 1)인 경우의 예시이다.

배열 A에 (3, 4, 2), (4, 2, 1) 순서로 연산을 수행하면 배열 A의 값은 12, (4, 2, 1), (3, 4, 2) 순서로 연산을 수행하면 15 이다.

배열 A와 사용 가능한 회전 연산이 주어졌을 때, 배열 A의 값의 최솟값을 구해보자. 회전 연산은 모두 한 번씩 사용해야 하며, 순서는 임의로 정해도 된다.

입력:

첫째 줄에 배열 A의 크기 N, M, 회전 연산의 개수 K가 주어진다.

둘째 줄부터 N개의 줄에 배열 A에 들어있는 수 A[i][j]가 주어지고, 다음 K개의 줄에 회전 연산의 정보 r, c, s가 주어진다.

출력:

배열 A의 값의 최솟값을 출력한다.

제한:

  • 3 ≤ N, M ≤ 50
  • 1 ≤ K ≤ 6
  • 1 ≤ A[i][j] ≤ 100
  • 1 ≤ s
  • 1 ≤ r-s < r < r+s ≤ N
  • 1 ≤ c-s < c < c+s ≤ M

풀이방법:

 제시되어 있는 방법대로 구현을 하면 되는 문제다. 다만 회전을 하는 것을 한번 구현한 뒤, 점점 안으로 작게 회전을 해야하고, 모서리 부분을 신경쓰면서 구현을 진행해야 한다.

 점점 작아지는 것은 while과 s를 사용해서 구현할 수 있다. s에 의해서 꼭짓점들이 구해지기 때문에, s를 1씩 빼면서 while 반복문을 사용한다. 종료되는 조건은 가장 왼쪽 윗 칸과 가장 오른쪽 아랫 칸이 작아지는 시점이며, r+s==r-s인 경우다.

 permutation을 사용해서 모든 조건을 구한 뒤에 입력받은 A를 copy한 뒤에 회전 연산을 해서 원래 입력값을 보존하도록 한다.

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
37
38
39
40
41
42
from itertools import permutations
import copy
 
def rotate(coord):
    r, c, s = coord    
    while True:
        if r-s==r+s:
            break
        tmp = None
        for i in range(c-s,c+s+1):
            if tmp is None:
                tmp = A[r-s][i]
            else:
                A[r-s][i], tmp = tmp, A[r-s][i]
        for i in range(r-s+1,r+s+1):
            A[i][c+s], tmp = tmp, A[i][c+s]
        for i in range(c+s-1,c-s-1,-1):
            A[r+s][i], tmp = tmp, A[r+s][i]
        for i in range(r+s-1,r-s-1,-1):
            A[i][c-s], tmp = tmp, A[i][c-s]
        s-=1
    
N, M, K = map(int,input().split())
= []
for _ in range(N):
    A.append(list(map(int,input().split())))
    
commands = []
for _ in range(K):
    r, c, s = map(int,input().split())
    commands.append((r-1, c-1, s))
    
command_lists = list(permutations(commands,K))
answer = float('inf')
for command_list in command_lists:
    A_copy = copy.deepcopy(A)
    for c in command_list:
        rotate(c)
    for i in range(N):
        answer = min(sum(A[i]),answer)
    A = copy.deepcopy(A_copy)
print(answer)    
cs

문제링크:

https://www.acmicpc.net/problem/17406

 

17406번: 배열 돌리기 4

크기가 N×M 크기인 배열 A가 있을때, 배열 A의 값은 각 행에 있는 모든 수의 합 중 최솟값을 의미한다. 배열 A가 아래와 같은 경우 1행의 합은 6, 2행의 합은 4, 3행의 합은 15이다. 따라서, 배열 A의

www.acmicpc.net

 

728x90
반응형

'Algorithm > Python' 카테고리의 다른 글

[BOJ]1655. 가운데를 말해요  (0) 2021.09.10
[BOJ]2110. 공유기 설치  (0) 2021.09.09
[BOJ]2638. 치즈  (0) 2021.09.06
[BOJ]2636. 치즈  (0) 2021.09.03
[BOJ] 11054. 가장 긴 바이토닉 부분 수열  (0) 2021.09.02

+ Recent posts