a, b = map(int, input().split())n = int(input())print(sum([(a+b)-i%(a+b) for i in list(map(int, input().split())) if i%(a+b) >= a]))
首先map() sum()傳入的都是iterable 所以多括一層list()和[]只是在浪費時間和記憶體轉成list (但list的常數很好 有時這樣會比較快 但絕對比較耗記憶體)
然後題解不應該直接貼程式碼 應該敘述解題邏輯
最後 明明就可以一行
print((lambda a, b, n: sum(a + b - i % (a + b) for i in map(int, input().split()) if i % (a + b) >= a))(*map(int, input().split()), input()))