Swift中如何将整数数组视为超大整数进行右位移操作？

要实现把整数数组当作连续位块组成的超大整数右移1位（左侧补0），核心逻辑是让每个元素的高位被前一个元素的最低位填充，第一个元素高位补0。以下是基于NumPy的高效实现方案：

核心思路 #

假设数组中每个元素是bit_width位的无符号整数：

1. 对每个元素单独右移1位，得到基础结果；
2. 提取前一个元素的最低位，左移bit_width-1位后，作为当前元素的最高位；
3. 将上述两个结果按位或，得到最终每个元素的值。

代码实现 #

import numpy as np

def shift_array_as_bigint(arr, bit_width=8):
    # 基础右移1位的结果
    shifted = arr >> 1
    # 生成前一个元素最低位左移后的进位值，首元素补0
    carry = np.concatenate([[0], (arr[:-1] & 1) << (bit_width - 1)])
    # 合并基础结果与进位值
    return shifted | carry

# 测试示例
input_arr = np.array([1, 2, 3], dtype=np.uint8)
output_arr = shift_array_as_bigint(input_arr)
print([bin(val) for val in output_arr])
# 输出：['0b0', '0b10000001', '0b1']

说明 #

• 该方案使用NumPy矢量化操作，避免Python循环，在处理大数组时效率极高；
• bit_width参数可根据实际元素的位宽调整（比如16位则设为16，移位位数改为15）；
• 使用无符号数据类型（如uint8）可避免有符号整数右移时的符号位补位问题。

内容的提问来源于stack exchange，提问作者lancylot2004