[Python] SWEA D3 : 1209 - [S/W 문제해결 기본] 2일차 - Sum

https://swexpertacademy.com/main/code/problem/problemDetail.do?problemLevel=3&contestProbId=AV13_BWKACUCFAYh&categoryId=AV13_BWKACUCFAYh&categoryType=CODE&problemTitle=&orderBy=RECOMMEND_COUNT&selectCodeLang=ALL&select-1=3&pageSize=10&pageIndex=1

SW Expert Academy

---

📌 작성한 코드 

for test in range(1,11):
    n = int(input())
    arr = [[int(x) for x in input().split()] for _ in range(100)]
    total = []
    
    for i in arr:
        total.append(sum(i))
    
    turn_arr = list(map(list,zip(*arr[::-1])))
    for i in turn_arr:
        total.append(sum(i))
    
    temp = 0
    for i in range(100):
        temp += arr[i][i]
    total.append(temp)

    temp = 0
    for i in range(100):
        temp += turn_arr[i][i]
    total.append(temp)

    print(f'#{test} {max(total)}')

 

 

📌 풀이 

1. 행의 합 구하기 

- 현재 배열에서 각 행을 sum으로 합을 구해서 저장 

for i in arr:
    total.append(sum(i))

 

2. 열의 합 구하기 

- 현재 배열을 90도 회전시켜서 다시 행의 합(처음 배열에서는 열) 구하기

turn_arr = list(map(list,zip(*arr[::-1])))
for i in turn_arr:
    total.append(sum(i))

 

3. 대각선 구하기

- 원래 배열과 회전한 배열에서 x,y인덱스가 같은 경우를 모아 합 구하기 

temp = 0
for i in range(100):
    temp += arr[i][i]
total.append(temp)

temp = 0
for i in range(100):
    temp += turn_arr[i][i]
total.append(temp)

 

4. 합의 최댓값 출력하기 

print(f'#{test} {max(total)}')