C++で3で割り切れるサイズ2または3のグループの数

数の配列が与えられた場合、3で割り切れるサイズ2と3のグループの数を見つける必要があります。2と3の組み合わせ数の合計を取得し、それらが3で割り切れるかどうかを確認できます。

例を見てみましょう。

入力

arr = [1, 2, 3, 4]

出力

4

3で割り切れる4つの組み合わせがあります。組み合わせは...

[1, 2]
[2, 4]
[1, 2, 3]
[2, 3, 4]

アルゴリズム

• アレイを初期化します。

• サイズ2のすべての組み合わせを取得するには、2つのループを記述します。

  • 各グループの合計を計算します。

  • 合計が3で割り切れる場合は、カウントをインクリメントします。

• サイズ3のすべての組み合わせを取得するには、3つのループを記述します。

  • 各グループの合計を計算します。

  • 合計が3で割り切れる場合は、カウントをインクリメントします。

• カウントを返します。

実装

以下は、C++での上記のアルゴリズムの実装です

#include <bits/stdc++.h>
using namespace std;
int getNumberOfGroupsDivisibleBy3(int arr[], int n) {
   int count = 0;
   for (int i = 0; i < n; i++) {
      for (int j = i + 1; j < n; j++) {
         int sum = arr[i] + arr[j];
         if (sum % 3 == 0) {
            count += 1;
         }
      }
   }
   for (int i = 0; i < n; i++) {
      for (int j = i + 1; j < n; j++) {
         for (int k = j + 1; k < n; k++) {
            int sum = arr[i] + arr[j] + arr[k];
               if (sum % 3 == 0) {
                  count += 1;
               }
         }
      }
   }
   return count;
}
int main() {
   int arr[] = { 2, 3, 4, 5, 6, 1, 2, 4, 7, 8 };
   int n = 10;
   cout << getNumberOfGroupsDivisibleBy3(arr, n) << endl;
   return 0;
}

出力

上記のコードを実行すると、次の結果が得られます。

57