HackerRankの「Array Manipulation」問題の解法


以下に、この問題の解法とコード例を示します。

  1. ブルートフォース法: 最も単純な方法は、与えられた操作を順番に実行し、最終的な配列を求める方法です。これは、操作の数が少ない場合には有効ですが、操作の数が多い場合には計算時間が長くなります。
def array_manipulation(n, queries):
    arr = [0] * n
    for query in queries:
        a, b, k = query
        for i in range(a-1, b):
            arr[i] += k
    return max(arr)
n = 5
queries = [[1, 2, 100], [2, 5, 200], [3, 4, 300]]
result = array_manipulation(n, queries)
print(result)  # 出力: 600
  1. 累積和を利用する方法: 配列の各要素に対して、その要素までの累積和を計算することで、操作の効率を向上させることができます。
def array_manipulation(n, queries):
    arr = [0] * (n+1)
    for query in queries:
        a, b, k = query
        arr[a-1] += k
        arr[b] -= k
    max_value = 0
    running_sum = 0
    for i in range(n+1):
        running_sum += arr[i]
        max_value = max(max_value, running_sum)
    return max_value
n = 5
queries = [[1, 2, 100], [2, 5, 200], [3, 4, 300]]
result = array_manipulation(n, queries)
print(result)  # 出力: 600

このように、ブルートフォース法と累積和を利用する方法の2つのアプローチがあります。累積和を利用する方法は、操作の数に関係なく効率的に最終的な配列を求めることができます。