描述
给 n 个非负整数,请你求出最大或和,最小或和,最大与和,最小与和这四个数之和。
最大或和为在 n 个数中,任取若干个数(不能不取),进行或运算后最大的数。
最小或和为在 n 个数中,任取若干个数(不能不取),进行或运算后最小的数。
最大与和为在 n 个数中,任取若干个数(不能不取),进行与运算后最大的数。
最小与和为在 n 个数中,任取若干个数(不能不取),进行与运算后最小的数。
1 <= n <= 1000000,0 <= nums[i] <= 2^32 - 1。
样例
给出 n = 3, nums = [1, 2, 3],返回 7。
解释:
最大或和:3,最小或和:1,最大与和:3,最小与和:0。
答案:3 + 1 + 3 + 0 = 7。
给出 n = 3, nums = [0, 0, 1],返回 2。
解释:
最大或和:1,最小或和:0,最大与和:1,最小与和:0。
答案:1 + 0 + 1 + 0 = 2。
给出 n = 5, nums = [12313, 156, 4564, 212, 12],返回25090。
解释:
最大或和:12765,最小或和:12,最大与和:12313,最小与和:0。
答案:12765 + 12 + 12313 = 25090。
给出 n = 3, nums = [111111, 333333, 555555],返回 1588322。
解释:
最大或和:917047,最小或和:111111,最大与和:555555,最小与和:4609。
答案:917047+ 111111+ 555555+ 4609 = 1588322。
思路
#ifndef C782_H
#define C782_H
#include<iostream>
#include<vector>
#include<cmath>
using namespace std;
class Solution {
public:
/**
* @param n:
* @param nums:
* @return: return the sum of maximum OR sum, minimum OR sum, maximum AND sum, minimum AND sum.
*/
long long getSum(int n, vector<int> &nums) {
// write your code here
if (n == 1)
return nums[0] * 4;
int Max = nums[0], Min = nums[0];// 最大与和为数组元素的最大值,最小或和为数组元素的最小值
long long res = 0;
vector<int> MinAnd(32,1),MaxOr(32,0);//构建MinAnd和MaxOr分别表示最小与和以及最大或和
//对于元素nums[j],在MinAnd中,若其第i位为0,则MinAnd[i]=0;在MaxOr中,若其第i位为1,则MaxOr[i]=1
for (int j = 0; j < nums.size();++j)
{
int temp = nums[j];
for (int i = 31; i >= 0; --i)
{
if (temp == 0)
{
MinAnd[i] = 0;
}
else
{
if (temp % 2 == 1)
MaxOr[i] = 1;
else
MinAnd[i] = 0;
}
temp /= 2;
}
if (Max < nums[j])
Max = nums[j];
if (Min > nums[j])
Min = nums[j];
}
int MinAndVal = 0, MaxOrVal = 0;
for (int i = 0; i < 32; ++i)
{
MinAndVal += MinAnd[i] * pow(2, 31 - i);
MaxOrVal += MaxOr[i] * pow(2, 31 - i);
}
res = Max + Min + MinAndVal + MaxOrVal;
return res;
}
};
#endif