题解 | LeetGPU Solutions

Contents

• Reverse Array

Reverse Array

LeetGPU

分析

本题注意处理一半 idx 即可，因为对所有 idx 进行处理会导致每组元素被交换两次，最终等价于未交换

Code

__global__ void reverse_array(float* input, int N) {
    int idx = blockDim.x * blockIdx.x + threadIdx.x;
    if (idx < N / 2) {
        float tmp = input[idx];
        input[idx] = input[N - idx - 1];
        input[N - idx - 1] = tmp;
    }
}