[python] SWEA 1209 - Sum

Problem

다음 100X100의 2차원 배열이 주어질 때, 각 행의 합, 각 열의 합, 각 대각선의 합 중 최댓값을 구하는 프로그램을 작성하여라.
다음과 같은 5X5 배열에서 최댓값은 29이다.

[제약 사항]
총 10개의 테스트 케이스가 주어진다.
배열의 크기는 100X100으로 동일하다.
각 행의 합은 integer 범위를 넘어가지 않는다.
동일한 최댓값이 있을 경우, 하나의 값만 출력한다.
 
[입력]
각 테스트 케이스의 첫 줄에는 테스트 케이스 번호가 주어지고 그 다음 줄부터는 2차원 배열의 각 행 값이 주어진다.

[출력]
#부호와 함께 테스트 케이스의 번호를 출력하고, 공백 문자 후 테스트 케이스의 답을 출력한다.

 

input.txt

1
13 24 13 24 1 7 24 11 22 18 22 16 24 8 15 28 9 24 14 14 28 18 17 9 3 29 22 12 28 2 25 6 11 26 14 19 3 26 13 6 23 3 3 29 13 25 4 27 8 25 28 8 9 17 28 13 24 27 9 25 21 20 6 16 28 5 22 11 9 29 13 26 28 2 11 10 14 14 5 11 26 9 15 3 23 9 8 11 12 6 9 18 6 14 28 21 24 24 20 12
20 28 29 21 27 13 29 16 6 28 5 7 13 20 7 1 11 1 23 12 4 9 27 19 26 2 21 2 1 18 4 20 6 4 18 9 20 3 28 28 1 21 1 2 11 7 20 15 7 29 14 7 15 10 29 24 2 25 29 3 11 9 17 6 2 17 17 11 7 20 26 10 8 1 15 10 2 29 7 9 17 8 25 28 29 12 28 19 3 4 17 17 28 9 2 15 14 6 20 3
13 19 28 23 11 1 19 5 5 28 19 18 16 14 26 20 6 12 20 18 16 22 25 21 25 29 3 ............
........
........

 

output.txt

#1 1712
#2 1743
.......
.......

Solve

100X100 데이터 받아오기 data = [list(map(int,input().split())) for _ in range (100)]

행, 열, 대각선 각각 합한후 sum 초기화

합 구할때마다 max 에 저장 후 비교

대각선 | -> data[i][i]

대각선 / -> data[i][99-i]

Code

for tc in range(1,11):
    num = int(input())
    data = [list(map(int, input().split())) for i in range(100)]
    max = 0
    
#행
    for y in range(100):
        sum = 0
        for x in range(100):
            sum += data[y][x]
        if sum > max:
            max = sum
            
#열
    for x in range(100):
        sum = 0
        for y in range(100):
            sum += data[y][x]
        if sum > max:
            max = sum
            
#대각선|
    sum = 0
    for i in range(100):
        sum += data[i][i]
        if sum > max:
            max = sum
            
#대각선/
    sum = 0
    for i in range(100):
        sum += data[i][99-i]
        if sum > max:
            max = sum
            
    print(f"#{tc} {max}")

 

# 본 문제의 저작권은 SW Expert 아카데미에 있습니다.

https://swexpertacademy.com/main/code/userProblem/userProblemDetail.do?contestProbId=AYU0Fmz6YwIDFAXn

SW Expert Academy