128. [삼성기출24]Python 백준 17140 번 이차원 배열과 연산

안녕하세요 코딩하는 덕구입니다.

백준 17140 이차원 배열과 연산 입니다.

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

17140번: 이차원 배열과 연산

 

먼저 백준 17140 번 이차원 배열과 연산 코드입니다. 설명은 아래에 있습니다.

from collections import Counter
from functools import reduce
r, c, k = map(int, input().split())
A = []
for _ in range(3):
    A.append(list(map(int, input().split())))

def R():
    max_num = 0
    for i in range(len(A)):
        arr = Counter(A[i])
        del arr[0]
        arr = list(arr.items())
        arr.sort(key=lambda x: (x[1], x[0]))
        if len(arr) > 50:
            arr = arr[:50]
        A[i] = list(reduce(lambda x, y: x + y, arr[1:], arr[0]))
        max_num = max(max_num, len(A[i]))

    for i in range(len(A)):
        if len(A[i]) < max_num:
            A[i].extend([0] * (max_num - len(A[i])))

r -= 1
c -= 1
time = -1
for cnt in range(0, 101):
    if r < len(A) and c < len(A[0]):
        if A[r][c] == k:
            time = cnt
            break

    if len(A) >= len(A[0]):
        R()
    else:
        A = list(zip(*A))
        R()
        A = list(zip(*A))


print(time)

 

문제 접근

배열의 r행 c열이  k 가 아니면 행과 열의 길이에 따라서 R 또는 C 연산을 합니다. 

이때 R 연산만 구현하고, C연산시에는 배열을 전치하여 R연산, 이후 다시 배열을 전치하면 됩니다.

 

Python 코드 구현

 필요한 라이브러리를 불러온 후 배열 A를 입력받습니다.

from collections import Counter
from functools import reduce
r, c, k = map(int, input().split())
A = []
for _ in range(3):
    A.append(list(map(int, input().split())))

 

0부터 100까지 반복하며 배열의 r행 c열이  k 가 아니면 R 또는 C 연산을 합니다. 

100을 넘어서도 불가능하다면 -1을 출력합니다.

r -= 1
c -= 1
time = -1
for cnt in range(101):
    if r < len(A) and c < len(A[0]):
        if A[r][c] == k:
            time = cnt
            break

    if len(A) >= len(A[0]):
        R()
    else:
        A = list(zip(*A))
        R()
        A = list(zip(*A))

print(time)

 

다음은 함수 R에 대한 설명입니다.

def R():
    max_num = 0
    for i in range(len(A)): #모든 배열에 접근
        arr = Counter(A[i]) #각 성분이 몇개 나왔는지 count
        del arr[0] #0은 제외
        arr = list(arr.items())
        arr.sort(key=lambda x: (x[1], x[0])) #배열을 새로운 정렬 기준에 맞추어 정렬
        if len(arr) > 50:
            arr = arr[:50]
        A[i] = list(reduce(lambda x, y: x + y, arr[1:], arr[0])) #리스트로 변환
        max_num = max(max_num, len(A[i]))

    for i in range(len(A)): #빈공간 0으로 채움
        if len(A[i]) < max_num:
            A[i].extend([0] * (max_num - len(A[i])))

 

파이썬 백준 17140 번 이차원 배열과 연산 최종 코드입니다. 

from collections import Counter
from functools import reduce
r, c, k = map(int, input().split())
A = []
for _ in range(3):
    A.append(list(map(int, input().split())))

def R():
    max_num = 0
    for i in range(len(A)): #모든 배열에 접근
        arr = Counter(A[i]) #각 성분이 몇개 나왔는지 count
        del arr[0] #0은 제외
        arr = list(arr.items())
        arr.sort(key=lambda x: (x[1], x[0])) #배열을 새로운 정렬 기준에 맞추어 정렬
        if len(arr) > 50:
            arr = arr[:50]
        A[i] = list(reduce(lambda x, y: x + y, arr[1:], arr[0])) #리스트로 변환
        max_num = max(max_num, len(A[i]))

    for i in range(len(A)): #빈공간 0으로 채움
        if len(A[i]) < max_num:
            A[i].extend([0] * (max_num - len(A[i])))

r -= 1
c -= 1
time = -1
for cnt in range(101):
    if r < len(A) and c < len(A[0]):
        if A[r][c] == k:
            time = cnt
            break

    if len(A) >= len(A[0]):
        R()
    else:
        A = list(zip(*A))
        R()
        A = list(zip(*A))

print(time)

 

백준 17140 번 이차원 배열과 연산 Python 코드였습니다.

감사합니다.