Question

我有一个pMat[M][N]形式的矩阵（其中M和N是可变的，因此是用户输入的）。我想使用内置的std::sort函数对二维数组的元素进行排序。

例如，考虑以下数组：

5 9 6
8 1 3
7 2 4

它应该输出为：

1 2 3
4 5 6
7 8 9

以下是我为此编写的代码：

#include <iostream>
#include <algorithm>

int main() {
    int M, N, **pMat;
    std::cin >> M >> N;
    pMat = new int* [M];
    for (int i=0; i<M; i++){
        pMat[i] = new int[N];
    }
    for (int i=0; i<M; i++){
        for (int j=0; j<N; j++){
            std::cin >> pMat[i][j];
        }
    }
    std::sort(&pMat[0][0], (&pMat[0][0])+(M*N));
    for (int i=0; i<M; i++){
        for (int j=0; j<N; j++){
            std::cout << pMat[i][j] << " ";
        }
        std::cout << std::endl;
    }
    return 0;
}

我基于以下理解编写了以上代码：

C++中的数组存储在连续的内存位置中
std::sort(add1,add2)对内存位置[add1,add2)中的所有元素进行排序

上面的代码没有给出正确的输出。例如，当提供以下输入时：

它显示以下输出：

我该如何编写代码？我的理解错在哪里？

（有关信息，我查看了以下答案：Sort a 2D array in C++ using built in functions(or any other method)?，但它没有回答我的查询）

Answer 1

一种方法是创建一个大的连续数组（可排序）的数组，然后通过第二个子数组访问该数组em>数组指针。

第二个数组仅包含一个指针列表，每个指针指向较大的指针中的不同子数组。

类似这样的东西：

int M, N;

std::cin >> M >> N;

// one big array of actual data
// (an array of contiguous sub-arrays)
std::vector<int> v(M * N);

// array of pointers to sub-arrays within the actual data
std::vector<int*> pMat;

// point the pointers at the actual data
// each pointer pointing to the relevant sub-array
for(int i = 0; i < M; i++)
    pMat.push_back(v.data() + (i * N));

// get the input, changing the actual data
// through the pointers
for(int i = 0; i < M; i++)
    for(int j = 0; j < N; j++)
        std::cin >> pMat[i][j];

// sort the actual data
std::sort(std::begin(v), std::end(v));

// look at the data through the sub-array pointers
for(int i = 0; i < M; i++)
{
    for(int j = 0; j < N; j++)
        std::cout << pMat[i][j] << " ";
    std::cout << '\n';
}

return 0;

注意：：我使用std::vector来管理我的数组，但是它也可以与使用new[]和delete[]创建的内置数组一起使用（不推荐）。

编辑：要添加。

或者（更好的是），您可以创建一个类，该类将数据存储在一个大的连续块中，并使用像这样的数学偏移量来访问不同的子数组：

template<typename T>
class two_dee_array
{
public:
    two_dee_array(std::size_t M, std::size_t N): v(M * N), stride(N) {}

    T& operator()(std::size_t M, std::size_t N)
        { return v[(M * stride) + N]; }

    T const& operator()(std::size_t M, std::size_t N) const
        { return v[(M * stride) + N]; }

    std::size_t col_size() const { return stride; }
    std::size_t row_size() const { return v.size() / stride; }

    auto begin() { return std::begin(v); }
    auto end() { return std::end(v); }

    auto begin() const { return std::begin(v); }
    auto end() const { return std::end(v); }

    auto cbegin() const { return std::cbegin(v); }
    auto cend() const { return std::cend(v); }

private:
    std::vector<int> v;
    std::size_t stride;
};

int main()
{
    int M, N;

    std::cin >> M >> N;

    two_dee_array<int> v(M, N);

    for(int i = 0; i < M; i++)
        for(int j = 0; j < N; j++)
            std::cin >> v(i, j);

    std::sort(std::begin(v), std::end(v));

    for(int i = 0; i < M; i++)
    {
        for(int j = 0; j < N; j++)
            std::cout << v(i, j) << " ";
        std::cout << '\n';
    }
}

Answer 2

正如@alter_igel所说。您的问题是具有非连续数组。在这里更正。

#include <iostream>
#include <algorithm>

int main() {

    int M, N, *pMat;
    std::cin >> M >> N;

    pMat = new int[M*N];

    for (int i=0; i<M; i++){
        for (int j=0; j<N; j++){
            std::cin >> pMat[i*N+j];
        }
    }
        std::cout << std::endl;

    std::sort(pMat, pMat+(M*N));

    for (int i=0; i<M; i++){
        for (int j=0; j<N; j++){
            std::cout << pMat[i*N+j] << " ";
        }
        std::cout << std::endl;
    }
    return 0;
}

使用内置的std :: sort函数

2 个答案: