最小の算術平均偏差を見つけるためのC++コード

3つの要素を持つ配列Aがあるとします。 A [1]は、A [0] + A [2] =2 * A [1]の場合、2つの要素A[0]とA[3]の算術平均です。 3つの数値d（A [0]、A [1]、A [2]）の算術平均偏差は| A [0] + A [2]-2 * A[1]|です。次の操作は何度でも実行できます。インデックス{0、1、2}からi！=jとなる2つのインデックスiとjを選択し、A [i]を1増やし、A[j]を1減らします。算術平均偏差の最小値を見つける必要があります。

したがって、入力がA =[2、2、6]の場合、出力は1になります。これは、A[0]を減らしてA[1]を増やすと、配列が[1、3、6]になるためです。したがって、平均偏差は| 1 + 6-2 *3|になります。 =1。

ステップ

これを解決するには、次の手順に従います-

a := A[0]
b := A[1]
c := A[2]
return minimum of 1 and (((a + c - 2 * b) mod 3 + 3) mod 3)

例

理解を深めるために、次の実装を見てみましょう-

#include <bits/stdc++.h>
using namespace std;
int solve(vector<int> A)
{
   int a = A[0];
   int b = A[1];
   int c = A[2];
   return min(1, ((a + c - 2 * b) % 3 + 3) % 3);
}
int main()
{
   vector<int> A = { 2, 2, 6 };
   cout << solve(A) << endl;
}

入力

{ 2, 2, 6 }

出力

1