列表

详情


3315. 构造最小位运算数组 II

给你一个长度为 n 的质数数组 nums 。你的任务是返回一个长度为 n 的数组 ans ,对于每个下标 i ,以下 条件 均成立:

除此以外,你需要 最小化 结果数组里每一个 ans[i] 。

如果没法找到符合 条件 的 ans[i] ,那么 ans[i] = -1 。

质数 指的是一个大于 1 的自然数,且它只有 1 和自己两个因数。

 

示例 1:

输入:nums = [2,3,5,7]

输出:[-1,1,4,3]

解释:

示例 2:

输入:nums = [11,13,31]

输出:[9,12,15]

解释:

 

提示:

原站题解

去查看

上次编辑到这里,代码来自缓存 点击恢复默认模板
class Solution { public: vector<int> minBitwiseArray(vector<int>& nums) { } };

golang 解法, 执行用时: 0 ms, 内存消耗: 3.3 MB, 提交时间: 2024-10-14 09:59:56

func minBitwiseArray(nums []int) []int {
	for i, x := range nums {
		if x == 2 {
			nums[i] = -1
		} else {
			nums[i] ^= (x + 1) &^ x >> 1
		}
	}
	return nums
}

java 解法, 执行用时: 1 ms, 内存消耗: 44 MB, 提交时间: 2024-10-14 09:59:43

class Solution {
    public int[] minBitwiseArray(List<Integer> nums) {
        int n = nums.size();
        int[] ans = new int[n];
        for (int i = 0; i < n; i++) {
            int x = nums.get(i);
            if (x == 2) {
                ans[i] = -1;
            } else {
                int invLb = (x + 1) & ~x;
                ans[i] = x ^ (invLb >> 1);
            }
        }
        return ans;
    }
}

cpp 解法, 执行用时: 0 ms, 内存消耗: 26.6 MB, 提交时间: 2024-10-14 09:59:31

class Solution {
public:
    vector<int> minBitwiseArray(vector<int>& nums) {
        for (int& x : nums) { // 注意这里是引用
            if (x == 2) {
                x = -1;
            } else {
                int inv_lb = (x + 1) & ~x;
                x ^= inv_lb >> 1;
            }
        }
        return nums;
    }
};

python3 解法, 执行用时: 55 ms, 内存消耗: 16.6 MB, 提交时间: 2024-10-14 09:59:01

class Solution:
    def minBitwiseArray(self, nums: List[int]) -> List[int]:
        for i, x in enumerate(nums):
            if x == 2:
                nums[i] = -1
            else:
                inv_lb = (x + 1) & ~x
                nums[i] ^= inv_lb >> 1
        return nums

上一题