C++学习对二维数组进行排序

Posted 2022-12-23 songyuc

1 问题描述

今天在学习 CUDA-NMS 算法时，遇到一个小问题，就是希望对一个二维数组进行排序，具体是这样的：

需要对存储目标框的二维数组进行排序，排序的方式是按照目标框的得分score从高往低排序，也就是结果数组的第二维的最后一个元素（dets[i][4]）

（1）尝试使用sort()直接进行排序，无法编译通过

刚开始听人说，可以用sort()直接进行排序，不过试了一下好像无法编译通过，示例代码：CPP_sort

#include <algorithm> // for std::sort

const int N = 3;
int arr[N][N] = 
        3, 1, 7,
        4, 1, 9,
        5, 3, 2,
    ;

int main()

    // Initialize the array with some values
    for (int i = 0; i < N; ++i)
        for (int j = 0; j < N; ++j)
            arr[i][j] = i * N + j;

    // Sort the array using a custom comparator function
    std::sort(arr, arr + N, [](const int a[], const int b[]) 
        return a[N-1] > b[N-1];
    );

    // The array is now sorted in descending order by the last element of the second dimension
    return 0;

我个人感觉这里面比较本质的原因是：二维数组（的一行）无法像一维数组的元素一样，在排序的时候进行swap交换，所以在编译的时候会报错；
我个人觉得比较看起来有关联的一句报错信息如下所示：

报错信息表示：error: invalid array assignment，很可能就是表示无法直接使用dets[i]的方式直接赋值数组的一行；

2 解决方法

代码示例：CPP_qsort_2d_array

Note:
这份代码示例没有用lambda表达式来书写，会使得代码的篇幅较长，之后会计划使用lambda表达式来进行改写。