Interleave Arrays

原始题目：LeetGPU - Interleave Arrays

题目描述

编写一个 GPU 程序，将两个 32 位浮点数数组交错合并。给定两个输入数组 $A$ 和 $B$，每个长度为 $N$，生成一个长度为 $2N$ 的输出数组，元素在 $A$ 和 $B$ 之间交替排列：

$$output = [A[0], B[0], A[1], B[1], A[2], B[2], \dots, A[N-1], B[N-1]]$$

实现要求

• 不允许使用外部库。
• solve 函数签名必须保持不变。
• 最终结果必须存储在 output 数组中。

示例

示例 1

Input:  A = [1.0, 2.0, 3.0], B = [4.0, 5.0, 6.0]
Output: [1.0, 4.0, 2.0, 5.0, 3.0, 6.0]

示例 2

Input:  A = [10.0, 20.0], B = [30.0, 40.0]
Output: [10.0, 30.0, 20.0, 40.0]

约束条件

• $1 \le N \le 50{,}000{,}000$。
• 性能测试在 $N = 25{,}000{,}000$ 的规模下进行。

解题思路

数组交错合并是一种分散写（Scatter）操作：连续读但间隔写。每个线程负责一对元素 $(A[i], B[i])$，将它们写入输出中的 $2i$ 和 $2i+1$ 位置。写入模式是规则的，只要线程按顺序写入即可实现合并访问。也可以使用 float2 向量化存储来一次写入两个元素，减少内存事务数。欢迎在 GitHub Discussions 分享你的解法。