毎朝ちびちびCodewars。
[6 kyu] Sums of Parts www.codewars.com
今日のお題：
"""
リストの各要素について、自分と自分より後ろの要素の和をリストにして返す
例: ls = [0, 1, 3, 6, 10]
自分より後ろの要素は、前から順に
[0, 1, 3, 6, 10] -> 和は 20
[1, 3, 6, 10] -> 和は 20
[3, 6, 10] -> 和は 19
[6, 10] -> 和は 16
[10] -> 和は 10
[] -> 和は 0
従って [20, 20, 19, 16, 10, 0] を返す
def Sums of Parts(ls):
Args:
ls (list): 整数のリスト
Returns:
(int): 和のリスト
"""
コード
mine
def parts_sums(ls): s, result = 0, [0] for x in ls[::-1]: s += x result.append(s) return result[::-1]
- リストをひっくり返して後ろから順に足していく方式。
- 最初に全要素の和を求めて、そこから一要素ずつ引く人が多かったけど、たぶん後ろから足す方が計算量が少ないから速い。
文法、関数・メソッドnote
itertools.accumulate(p [,func])
itertools --- 効率的なループ実行のためのイテレータ生成関数 — Python 3.7.4rc1 ドキュメント
- p0, p0+p1, p0+p1+p2, ...を返す。 例: accumulate([1,2,3,4,5]) --> 1 3 6 10 15
むむ、そのまま使えるモジュールがあるじゃないか ^^;
以下、Darshan97 （他ユーザ）のコード。簡潔で最速。 https://www.codewars.com/kata/reviews/5cff429c2ad4650001bf1590/groups/5cff4dd10310c200010520cc
from itertools import accumulate def parts_sums(ls): return [*accumulate(ls[::-1])][::-1]+[0]
- func（2つの引数を取る関数）を与えるとさまざまな累積結果が求められる
accumulate([1,2,3,4,5], operator.mul) --> 1 2 6 24 120 accumulate([3, 4, 6, 2, 1, 9, 0, 7, 5, 8], max)) --> 3 4 6 6 6 9 9 9 9 9
