Если вы имеете в виду
N = matrice.size() = no. rows of the matrix!
следующее должно делать эту работу, даже если это не лучший способ.
- Укажите структуру
ElementIntex
, в которой можно хранить элемент из matrix
и их соответствующих индексов .
- Перебирать элементы в
matrix
и сохранять их в векторе ElementIntex
.
- Сортировка
std::vector<ElementIntex>
по элементу в структуре ElementIntex
с использованием двоичного предиката. (В порядке убывания)
- Возвращает первое
N
количество элементов из этого отсортированного std::vector<ElementIntex>
, где N
равно no. строк в matrix
.
Ниже приведен пример кода: ( См. Live )
#include <iostream>
#include <vector>
#include <cstddef> // std::size_t
#include <algorithm> // std::sort
struct ElementIntex
{
std::size_t col, row;
double element;
ElementIntex(std::size_t cl, std::size_t rw, double ele)
: col{cl}
, row{rw}
, element{ele}
{}
};
std::vector<ElementIntex> getLargestElements(
const std::vector<std::vector<double>>& matrice)
{
std::vector<ElementIntex> vec;
// reserve the memory to prevent unwanted reallocations: if you now the size
// vec.reserve(/*total no. of elements*/)
std::size_t rowIndex = 0;
for (const std::vector<double>& row : matrice)
{
std::size_t colIndex = 0;
for (const double element : row)
vec.emplace_back(rowIndex, colIndex++, element);
++rowIndex;
}
// sort descending order of elements in the vector of `ElementIntex`
std::sort(vec.begin(), vec.end(),
[](const auto & lhs, const auto & rhs) { return lhs.element > rhs.element; });
// return N largest elements from the sorted vector: where N = matrice.size() = no. rows!
return { vec.cbegin(), vec.cbegin() + matrice.size() };
}
int main()
{
// consider the following vector of vectors(matrx in your case)
std::vector<std::vector<double>> matrice{
{1.05, -8.05, 1.0, 8.58, 3.04},
{15.05, 8.05, 7.05, 8.58},
{11.05, 88.05, 7.06},
{-12.05, -8.05}
};
const auto resultVec{ getLargestElements(matrice) };
for (const ElementIntex& elementIndex : resultVec)
std::cout << "The element " << elementIndex.element
<< " and index [" << elementIndex.row
<< "][" << elementIndex.col << "]\n";
return 0;
}
выход
The element 88.05 and index [1][2]
The element 15.05 and index [0][1]
The element 11.05 and index [0][2]
The element 8.58 and index [3][0]