def solution(arr1, arr2): ans = arr1 for i in range(len(arr1)): for j in range(len(arr1[i])): ans[i][j] = arr1[i][j]+arr2[i][j] return ans
가장 간단한 방법인 for문을 사용해 풀었다.
풀면서도 이건... 시간복잡도가 높을거라고 생각했다.
zip()
함수를 사용할까 했는데, 어떻게 써야 문제를 풀 수 있는지를 모르겠더라.....
역시나 잘한 풀이는 zip()
을 사용한 풀이었다.
나는 zip()
을 한번만 사용하고 그 안에서 다시 for문을 사용하면 되나..? 했는데, 그냥 zip()
을 2번 사용하면 되는거였다..!!!!!!!!
zip()
이랑 리스트컴프리핸션을 2개 겹쳐서 사용하는 방법좀 공부해야겠다.....