improved memory usage

Author: AdarshGupta-dev

Divisibility-by-K.py

@@ -67,10 +67,8 @@ def solve(arr, length, K):
     length, K = input().split()
     K = int(K)
 
-    arr = input()
-    arr = list(map(int, arr.split()))
+    arr = list(map(int, input().split()))
 
     ans.append(solve(arr, length, K))
 
-for i in ans:
-    print(i)
+print(*ans, sep="\n")