[백준/Python] 17140번(구현)_이차원 배열과 연산

📌문제 유형

구현, 시뮬레이션

 

📌문제

17140번: 이차원 배열과 연산

 

📌나의 문제풀이

- R연산, C연산 따로

• C연산의 경우 list(zip(*A))로 행,열 바꿔서 계산
• Counter 함수 이용해서 열 내의 숫자 쉽게 구할 수 있음! {숫자 : 숫자의 개수 count}
• A[r][c]에서 r,c 위치가 매번 있는것은 아님 -> 예외처리 필요

from collections import Counter

r,c,k = map(int,input().split())
r,c = r-1, c-1

A = []
for _ in range(3):
    A.append(list(map(int,input().split())))

ans = 0

def r_cal(A):
    matrix = []
    max_len, cnt_rows = 0, 0

    for i in range(len(A)):
        new_list = []
        cnt_rows += 1
        for num, cnt in Counter(A[i]).items():
            # 단, 0의 개수는 제외
            if num != 0:
                new_list.append([num, cnt])
        
        # 등장 횟수 커지는 순, 수가 커지는 순
        new_list.sort(key = lambda x:(x[1],x[0]))
        # 100개 전까지
        if cnt_rows <= 100:
            matrix.append(sum(new_list,[]))
            max_len = max(len(sum(new_list,[])),max_len)

    for i in range(len(matrix)):
        if len(matrix[i]) < max_len:
            matrix[i].extend([0 for _ in range(max_len - len(matrix[i]))])

    A = matrix
    return A

def c_cal(A):
    # 행,렬 바꾸기
    A = list(zip(*A))

    matrix = []
    max_len, cnt_rows = 0, 0

    for i in range(len(A)):
        new_list = []
        cnt_rows += 1
        for num, cnt in Counter(A[i]).items():
            # 단, 0의 개수는 제외 - 구현하기
            if num != 0:
                new_list.append([num, cnt])
        
        
        # 등장 횟수 커지는 순, 수가 커지는 순
        new_list.sort(key = lambda x:(x[1],x[0]))
        # 100개 전까지
        if cnt_rows <= 100:
            matrix.append(sum(new_list,[]))
            max_len = max(len(sum(new_list,[])),max_len)

    for i in range(len(matrix)):
        if len(matrix[i]) < max_len:
            matrix[i].extend([0 for _ in range(max_len - len(matrix[i]))])

    matrix = list(zip(*matrix))
    A = matrix
    return A
    


while True:
    # 위치가 항상 있는건 아님!
    if 0 <= r < len(A) and 0 <= c < len(A[0]):
        if A[r][c] == k:
            print(ans)
            break

    if ans > 100:
        print(-1)
        break

    ans += 1
    if len(A) >= len(A[0]):
        A = r_cal(A)
    else:
        A = c_cal(A)

 

📌다른사람의 문제풀이

- R,C는 행렬 회전 유무 빼고 코드 동일 -> 함수에 R,C 중 어떤 연산인지 선택하며 하나의 함수로 만들기

import sys
r,c,k = map(int, sys.stdin.readline().split())
A = [list(map(int, sys.stdin.readline().split())) for _ in range(3)]
def sort_new(matrix, RC):
    sorted_matrix = []
    max_count = 0
    for i in range(len(matrix)):
        B = []
        dic = dict()
        for j in range(len(matrix[i])):
            if matrix[i][j]!=0:
                if matrix[i][j] not in dic:
                    dic[matrix[i][j]] = 1
                else: dic[matrix[i][j]] += 1
        for key,value in dic.items():
            B.append([key,value])
        B.sort(key=lambda x: [x[1],x[0]])
        C = sum(B,[])
        max_count = max(max_count,len(C))
        sorted_matrix.append(C)
    for i in sorted_matrix:
        i+=[0]*(max_count-len(i)) # 가장 긴 길이에 맞춰서 0 추가
        if len(i)>100:
            i = i[:100]
    return sorted_matrix if RC=="R" else list(zip(*sorted_matrix))

result = 0
while True:
    count = 0
    if result >=101:
        result = -1
        break
    if r-1<len(A) and c-1<len(A[0]):
        if A[r-1][c-1] == k:
            break
    if len(A)>=len(A[0]):
        #R연산
        A = sort_new(A,"R")
    else :
        # C연산
        A = sort_new(list(zip(*A)),"C")
    result += 1

print(result)

 

📌리뷰

- 90도 회전 : list(zip(*A[::-1))

- 행과 열 바꾸기 : list(zip(*A))

python__[2차원 리스트 90도 회전, 행과 열 바꾸기]