[leetCode] 1672. Richest Customer Wealth (Python)

 ▼ 문제 바로가기 (링크) ▼

https://leetcode.com/problems/richest-customer-wealth/

Richest Customer Wealth - LeetCode

---

정수로 이루어진 m x n  의 그리드 행렬이 주어진다.

[i][j] 은 i번째 손님이 j은행에 가지고 있는 재산의 양이다.

가장 부유한 손님의 재산을 return하라.

손님의 재산은 모든 은행의 자산을 합친 값이다.

 

문제만 읽었을 때는 바로 와닿지 않았는데 예시와 제한 사항을 보면 이해가 쉽다.

 

m은 리스트의 길이, n은 내부 리스트의 길이.

따라서 내부 리스트의 원소들을 전부 더하고  그 중 가장 큰 값을 return하면 된다.

---

accounts = [[1, 5], [7, 3], [3, 5]]

class Solution:
    def maximumWealth(self, accounts):
        tmp = []
        for i in accounts:
            tmp.append(sum(i))
        return max(tmp)

s = Solution()
print(s.maximumWealth(accounts))

처음에는 이중 for문을 돌면서 두 번째 for문에서 

내부리스트 합산을 마치고 대소 비교를 하는 무식한 방법으로 만드려다가

잘 안 돼서 검색해서 풀었다. 이미 알고 있던 함수인데도 생각을 못했다.

sum() ,  max() 로 간단하게 해결할 수 있다.