C++でKビットが異なる配列のすべてのペアをカウントします

このチュートリアルでは、Kビットが異なる配列のペアの数を見つけるプログラムについて説明します。

このために、配列と整数Kが提供されます。私たちのタスクは、バイナリ表現でKビットが異なるペアの数を見つけることです。

例

#include <bits/stdc++.h>
using namespace std;
//counting number of bits in
//binary representation
int count_bit(int n){
   int count = 0;
   while (n) {
      if (n & 1)
         ++count;
      n >>= 1;
   }
   return count;
}
//counting the number of pairs
long long count_pair(int arr[], int n, int k) {
   long long ans = 0;
   for (int i = 0; i < n-1; ++i) {
      for (int j = i + 1; j < n; ++j) {
         int xoredNum = arr[i] ^ arr[j];
         if (k == count_bit(xoredNum))
            ++ans;
      }
   }
   return ans;
}
int main() {
   int k = 2;
   int arr[] = {2, 4, 1, 3, 1};
   int n = sizeof(arr)/sizeof(arr[0]);
   cout << "Total pairs for k = " << k << " are " << count_pair(arr, n, k) << "\n";
   return 0;
}

出力

5