Count 2D Array Element

原始题目：LeetGPU - Count 2D Array Element

题目描述

统计 $N \times M$ 的 32 位整数二维数组中值为 $k$ 的元素数量。

实现要求

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

示例

Input:  input = [[1,2,3],[4,5,1]], k = 1
Output: 2

约束条件

• $1 \le N, M \le 10{,}000$，$1 \le input[i][j], k \le 100{,}000$。

解题思路

与一维版本相同：逐元素比较 + 规约求和。将二维索引 $(i, j)$ 线性化为一维索引 $(i \cdot M + j)$，每个线程比较一个元素，分块做 warp shuffle 规约求和。属于内存带宽受限内核。欢迎在 GitHub Discussions 分享你的解法。