在本教程中,我们将讨论一个程序,以查找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