计算在 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

更新于: 10-Feb-2020

125 条浏览量

开始您的职业生涯

完成课程即可获得认证

开始
广告