[python] SWEA - 1209. [S/W 문제해결 기본] 2일차 - Sum

🤔문제 해결

• 가로, 세로, 대각선으로 나눠서 해결
• 가로: 한 행씩 sum 해줌
• 세로: 90도 돌린다음 할까 하다가 굳이라는 생각에... 리스트를 만들어 한 열끼리 수를 더해줌
• 대각선: 정사각형이므로 n행n열, n행(N-1-n)열 로 하나씩 더해줌
• 가장 큰 값

💨 오랜만에 다시 시작해서 가벼운 문제로 시작(❁´◡`❁)

 

💻소스 코드

T = 10  # 테케는 10개

for test_case in range(1, T + 1):
    input()  # 테케 번호 받기
    N = 100
    answer = 0
    arr = [list(map(int, input()[:-1].split(' '))) for _ in range(N)]  # 2 차원 행렬화
    # [:-1]을 한 이유는 맨 마지막에 띄어쓰기가 들어간다. 오랜만에 해서 왜 그런지 모르겠음

    # 가로 합
    for i in range(N):
        answer = max(sum(arr[i]), answer)

    vertical_sum_list = [0] * len(arr[0])
    # 세로 합
    for i in range(N):
        for j in range(N):
            vertical_sum_list[j] += arr[i][j]

    answer = max(max(vertical_sum_list), answer)

    # 대각선
    right_cross_sum = 0
    for i in range(N):
        right_cross_sum += arr[i][i]
    answer = max(answer, right_cross_sum)

    left_cross_sum = 0
    for i in range(N):
        left_cross_sum += arr[i][N - 1 - i]
    answer = max(answer, left_cross_sum)

    print(f'#{test_case} {answer}')

📕문제 확인

출처: SW Expert Academy