SWEA: 1209 Sum

문제

SW Expert Academy

 

문제 풀이

출처 : SW Expert Academy

 위의 그림과 같이 행렬의 모든 합을 구하고, 최대 값을 반환하는 문제이다. 이는 파이썬의 경우 간단하게 해결할 수 있다.

 

trans_array = list(map(list, zip(*array)))

 위의 행렬을 `numpy`를 사용하지 않고 전치하기 위해서는 위와 같이 전치할 수 있다. 따라서 전치시키게 되면 행렬을 합을 보다 편하게 구할 수 있다.

 

# ↘ 대각선의 합
left_right = sum([array[i][i] for i in range(100)])
# ↙ 대각선의 합
right_left = sum([array[i][99 - i] for i in range(100)])

 대각선의 합은 좌측에서 우측으로 내려오는 경우 [0][0], [1][1]과 같이 접근하면 된다. 이와 달리 우측에서 좌측으로 내려오는 경우의 대각선은 [0][99], [1][98]과 같이 접근하면 된다.

 

코드

for test_case in range(1, 11):
    answer = 0
    t = int(input())

    array = [list(map(int, input().split())) for _ in range(100)]
    # 행렬 전치
    trans_array = list(map(list, zip(*array)))
    # ↘ 대각선의 합
    left_right = sum([array[i][i] for i in range(100)])
    # ↙ 대각선의 합
    right_left = sum([array[i][99 - i] for i in range(100)])

    for i in range(100):
        answer = max(answer, max(sum(array[i]), sum(trans_array[i])))

    answer = max(answer, max(left_right, right_left))

    print('#' + str(test_case), answer)