C++で自然数からいくつかの整数を削除した後のK番目に小さい要素

このチュートリアルでは、自然数からいくつかの整数を削除した後、最小の要素を見つけるプログラムを作成します。

要素の配列とk値を指定しました。指定された配列に存在する自然数からすべての要素を削除します。そして、残りの自然数からk番目に小さい数を見つけます。

問題を解決するための手順を見てみましょう。

• 配列とkを初期化します。
• 配列を初期化し、指定された配列に存在する要素を除くすべての要素を0で初期化します。
• 指定された配列のサイズになるまで繰り返すループを記述します。
  • 現在の要素が上記の配列に存在しない場合は、kの値をデクリメントします。
  • kがゼロになったときに現在の値を返します。
• 0を返します。

例

コードを見てみましょう。

#include <bits/stdc++.h>
#define MAX 1000000
using namespace std;
int smallestNumber(int arr[], int n, int k) {
   int flag[MAX];
   memset(flag, 0, sizeof flag);
   for (int i = 0; i < n; i++) {
      flag[arr[i]] = 1;
   }
   for (int i = 1; i < MAX; i++) {
      if (flag[i] != 1) {
         k--;
      }
      if (!k) {
         return i;
      }
   }
   return 0;
}
int main() {
   int k = 2;
   int arr[] = { 3, 5 };
   cout << smallestNumber(arr, 2, k) << endl;
   return 0;
}

出力

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

2

結論

チュートリアルに質問がある場合は、コメントセクションにそのことを記載してください。