我正在尝试使用嵌套向量对多列进行排序,但我不确定如何实际执行。我在这里搜索了一堆帖子,但它们只显示了如何对最多两列进行排序,我知道对一列进行排序如下:
sort(myVector.begin(), myVector.end(), [](vector<int> const a, vector<int> const b){return a[0] < b[0];});
我有一个输入,用户输入向量的大小并相应地生成,我希望能够对输出的每一列进行排序。
例如:
未分类
{3, 7, 2}
{9, 6, 8}
{5, 1, 4}
已排序
{3, 1, 2}
{5, 6, 4}
{9, 7, 8}
这是线性代数库中的经典之作:由于元素访问的模式,给定矩阵的布局会对性能产生很大影响。
你面临着完全相同的结果。显然,在您的情况下,每个 int 向量都是一行。您需要对列进行排序。你可以做的是转置技巧:
一些代码来说明这个想法:
std::vector<std::vector<int>> transpose(std::vector<std::vector<int>> const& input) {
// For simplicity, I assume input is well formed, i.e.:
// - All int vectors have the same size
// - input is non-empty.
std::vector<std::vector<int>> tr_input;
for(std::size_t i = 0; i < input.front().size(); ++i) {
std::vector<int> tmp;
for (auto& vec : input) {
tmp.push_back(vec.at(i));
}
tr_input.push_back(tmp);
}
return tr_input;
}
现在我们有了转置函数,我们可以实现算法了。
std::vector<std::vector<int>> input = { { 3, 7, 2 },
{ 9, 6, 8 },
{ 5, 1, 4 } };
auto tr = transpose(input);
for (auto& v : tr) {
std::sort(v.begin(), v.end());
}
auto sorted = transpose(tr);
结果可以在此查看 Live Demo
本文收集自互联网,转载请注明来源。
如有侵权,请联系 [email protected] 删除。
我来说两句