Python
 Computer >> コンピューター >  >> プログラミング >> Python

分割統治法で最大部分配列問題を解くPythonプログラムの実装方法

本記事では、分割統治法(Divide and Conquer)を用いて最大部分配列問題(Maximum Subarray Problem)を解くPythonプログラムを紹介します。最大部分配列問題とは、配列内の連続する要素の合計が最大となる部分配列を見つける古典的なアルゴリズム問題です。

以下に、実際の実装例を示します。

サンプルコード

def max_crossing_sum(my_array, low, mid, high):

    sum_elements = 0
    sum_left_elements = -10000

    for i in range(mid, low-1, -1):
        sum_elements = sum_elements + my_array[i]

        if (sum_elements > sum_left_elements):
            sum_left_elements = sum_elements

    sum_elements = 0
    sum_right_elements = -1000
    for i in range(mid + 1, high + 1):
        sum_elements = sum_elements + my_array[i]

        if (sum_elements > sum_right_elements):
            sum_right_elements = sum_elements

    return max(sum_left_elements + sum_right_elements, sum_left_elements, sum_right_elements)

def max_sub_array_sum(my_array, low, high):

    if (low == high):
        return my_array[low]

    mid = (low + high) // 2

    return max(max_sub_array_sum(my_array, low, mid), max_sub_array_sum(my_array, mid+1, high), max_crossing_sum(my_array, low, mid, high))

my_list = [23, 12, 45, 67, 89, 11]
list_length = len(my_list)
print("The list is :")
print(my_list)

max_sum = max_sub_array_sum(my_list, 0, list_length-1)
print("The maximum contiguous sum is ")
print(max_sum)

出力結果

The list is :
[23, 12, 45, 67, 89, 11]
The maximum contiguous sum is
247

コードの解説

  • まず、「max_crossing_sum」という関数を定義します。この関数は、リストを中央で分割した際に境界をまたぐ部分配列の合計を計算します。

  • 続いて、「max_sub_array_sum」関数が再帰的に呼び出され、すべての部分配列の合計を効率的に計算します。

  • 関数の外側でリストを定義し、その内容をコンソールに表示します。

  • len() 関数を使用して、リストの長さを取得します。

  • 定義したリストとインデックス範囲を引数として渡し、最大部分配列の合計を計算する関数を呼び出します。

  • 最後に、計算された最大連続合計値をコンソールに出力します。

計算量について

分割統治法を用いたこのアプローチの時間計算量は O(n log n) です。これは、すべての部分配列を総当たりで調べる全探索(O(n²))よりも効率的な手法です。さらに、Kadaneのアルゴリズムを使えば O(n) で解くことも可能ですが、分割統治法はアルゴリズムの基礎を学ぶ上で非常に重要な考え方となります。

  1. Pythonで式木(式ツリー)を構築して評価するプログラムの実装方法

    はじめに本記事では、式木(Expression Tree)の後順巡回(後置記法・逆ポーランド記法)の結果が与えられたとき、そこから式木を復元(構築)し、さらにその式を評価して計算結果を求めるプログラムをPythonで実装します。最終的には、構築した式木の根(ルート)と、木全体を評価した値を返します。問題例次のような後置記法のトークン列が入力として与えられたとします。[1, 2, -, 3, 4, +, *]この列から式木を構築して評価すると、中間記法では (1 - 2) * (3 + 4) に相当し、計算結果は -7 になります。アルゴリズムの流れまず、子の接続位置を表す定数を定義しておきます

  2. Pythonで最大の成功確率を持つパスを見つけるプログラムの実装方法

    問題の概要 n 個のノード(ノードには 0 から順に番号が振られています)からなる無向重み付きグラフを考えます。このグラフは辺リスト(edge list)として入力され、各辺 e には「その辺を通過する際の成功確率」probability[e] が割り当てられています。さらに、開始ノード(start)と終了ノード(end)も与えられます。 求めたいのは、start から end へ移動するときに成功確率が最大となる経路であり、答えとしてその成功確率を返します。経路がひとつも存在しない場合は 0 を返してください。 たとえば、次のような入力が与えられたとします。 この場合の出力は 0.25