所有整数对之间的位差异数之和(Sum of bit differences among all pairs)

22 篇文章 0 订阅
3 篇文章 0 订阅

原文地址:Sum of bit differences among all pairs
已知一个有n个整数的数组,找出数组中任意两个元素组成的整数的位差异数的和。

例子:

输入: arr[] = {1, 2} 
输出: 4 
数组中所有的整数对:(1, 1), (1, 2) (2, 1), (2, 2) 
位差异数之和 = 0 + 2 + 2 + 0 = 4 

输入: arr[] = {1, 3, 5} 
输出: 8 
数组中所有的整数对:(1, 1), (1, 3), (1, 5) (3, 1), (3, 3) (3, 5), (5, 1), (5, 3), (5, 5) 
位差异数之和 = 0 + 1 + 1 + 1 + 0 + 2 + 1 + 2 + 0 = 8

一个简单的想法就是跑上两个循环逐个考虑所有的整数对,对于每一个整数对,计算位的差异。最后返回差异数的总和。这种方法的时间复杂度是O(n^2)。

有一个有效的方法可以在O(n)的时间内解决这个问题,它的方法就是将所有的数字都用32位(或者是其他固定位数)表示。这个想法是计数单独位的差异。我们可以遍历0到31并计数第i位的置位。设这个计数是“count”。这里将有“n - count”个数字的第i位没有置位,所以第i位的差异数是“count * (n - count) * 2”

下面是基于上述想法的C++实现:

// C++ program to compute sum of pairwise bit differences
#include <bits/stdc++.h>
using namespace std;

int sumBitDifferences(int arr[], int n)
{
    int ans = 0;  // Initialize result

    // traverse over all bits
    for (int i = 0; i < 32; i++)
    {
        // count number of elements with i'th bit set
        int count = 0;
        for (int j = 0; j < n; j++)
            if ( (arr[j] & (1 << i)) )
                count++;

        // Add "count * (n - count) * 2" to the answer
        ans += (count * (n - count) * 2);
    }

    return ans;
}

// Driver prorgram
int main()
{
    int arr[] = {1, 3, 5};
    int n = sizeof arr / sizeof arr[0];
    cout << sumBitDifferences(arr, n) << endl;
    return 0;
}

输出:

8
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值