在本教程中,我们将讨论一个程序,以查找K位不同的数组对的数量。
为此,我们将提供一个数组和一个整数K。我们的任务是查找在二进制表示形式中相差K位的对数。
#include <bits/stdc++.h>
using namespace std;
//的位数
//二进制表示
int count_bit(int n){
int count = 0;
while (n) {
if (n & 1)
++count;
n >>= 1;
}
return count;
}
//计算对数
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