Pythonで最も裕福な顧客の総資産額を求めるプログラム

本記事では、m × n の行列 accounts が与えられたとき、accounts[i][j] が「i 番目の顧客が j 番目の銀行に保有している金額」を表すものとして、最も裕福な顧客の総資産額を求める方法を解説します。

ここで「最も裕福な顧客」とは、すべての銀行における保有金額の合計が最大となる顧客のことです。

問題の例

例えば、入力が以下のような行列だったとします。

この場合、出力は 55 になります。2番目の顧客（2行目）の資産は 30 + 5 + 20 = 55 となり、全顧客の中で最大だからです。

解法のアプローチ

この問題は、次の手順で解くことができます。

• 最大値を保持する変数 max_value を 0 で初期化します。

• 各行（各顧客）ごとに、その行の全要素の合計値を計算します。

• 計算した合計値が現在の max_value より大きければ、max_value を更新します。

• すべての行を処理し終えたら、max_value を返します。これが最も裕福な顧客の総資産額です。

このアルゴリズムの計算量は、行列の全要素を一度ずつ走査するため O(m × n) であり、非常に効率的です。

Pythonでの実装例

理解を深めるために、以下の実装例を見てみましょう。

def solve(accounts):
    max_value = 0
    ind_value = 0
    for i in range(len(accounts)):
        ind_value = sum(accounts[i])
        if ind_value > max_value:
            max_value = ind_value
    return max_value

accounts = [[10,20,15],
    [30,5,20],
    [10,5,12],
    [15,12,3]]
print(solve(accounts))

入力

[[10,20,15],
[30,5,20],
[10,5,12],
[15,12,3]]

出力

55

補足：より簡潔な書き方

Pythonでは、内包表記と組み込み関数 max() を使うことで、上記のロジックを1行で表現することもできます。

def solve(accounts):
    return max(sum(row) for row in accounts)

どちらの実装でも結果は同じですが、コードの可読性やチームのコーディング規約に応じて使い分けるとよいでしょう。