C++の特定のサブアレイ内の特定の数以下の要素の数

番号とサブ配列の下限と上限のインデックスが与えられます。指定された数以下の要素の数を数える必要があります。例を見てみましょう。

入力

arr = [1, 2, 3, 4, 5, 6, 7, 8]
k = 4
lower = 0
upper = 5

出力

4

インデックス0と5の間には、4以下の4つの要素があります。

アルゴリズム

• 配列、番号、およびサブ配列のインデックスを初期化します。

• カウントを0に初期化します。

• サブアレイの下位インデックスからサブアレイの上位インデックスまで反復するループを記述します。

  • 現在の要素が指定された数以下の場合は、カウントを増やします。

• カウントを返します。

実装

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

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

出力

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

4
0