要么使用标头<numeric> 中声明的标准算法std::accumulate 和使用该算法并将传递给标准算法std::sort 的lambda 表达式,要么自己编写类似的函数。
以下是实现这两种方法的两个演示程序。
#include <iostream>
#include <iomanip>
#include <vector>
#include <iterator>
#include <algorithm>
#include <numeric>
#include <cstdlib>
#include <ctime>
int main()
{
size_t n = 0, m = 0;
std::cin >> n >> m;
std::vector<std::vector<int>> v( n, std::vector<int>( m ) );
std::srand( ( unsigned int )std::time( nullptr ) );
for ( auto &row : v )
{
for ( auto &item : row )
{
item = std::rand() % ( n * m );
}
}
for ( const auto &row : v )
{
for ( const auto &item : row )
{
std::cout << std::setw( 2 ) << item << ' ';
}
std::cout << '\n';
}
std::cout << '\n';
auto less = []( const auto &row1, const auto &row2 )
{
return std::accumulate( std::begin( row1 ), std::end( row1 ), 0ll ) <
std::accumulate( std::begin( row2 ), std::end( row2 ), 0ll );
};
std::sort( std::begin( v ), std::end( v ), less );
for ( const auto &row : v )
{
for ( const auto &item : row )
{
std::cout << std::setw( 2 ) << item << ' ';
}
std::cout << '\n';
}
std::cout << '\n';
return 0;
}
和
#include <iostream>
#include <iomanip>
#include <vector>
#include <iterator>
#include <algorithm>
#include <cstdlib>
#include <ctime>
long long int accumulate( const std::vector<int> &v, long long int init = 0 )
{
for ( const auto &item : v ) init += item;
return init;
}
bool less( const std::vector<int> &v1, const std::vector<int> &v2 )
{
return accumulate( v1 ) < accumulate( v2 );
}
int main()
{
size_t n = 0, m = 0;
std::cin >> n >> m;
std::vector<std::vector<int>> v( n, std::vector<int>( m ) );
std::srand( ( unsigned int )std::time( nullptr ) );
for ( auto &row : v )
{
for ( auto &item : row )
{
item = std::rand() % ( n * m );
}
}
for ( const auto &row : v )
{
for ( const auto &item : row )
{
std::cout << std::setw( 2 ) << item << ' ';
}
std::cout << '\n';
}
std::cout << '\n';
std::sort( std::begin( v ), std::end( v ), less );
for ( const auto &row : v )
{
for ( const auto &item : row )
{
std::cout << std::setw( 2 ) << item << ' ';
}
std::cout << '\n';
}
std::cout << '\n';
return 0;
}
如果输入的向量大小等于3 和4,则输出可能如下所示
3 3 1 4
6 1 5 7
5 6 7 2
3 3 1 4
6 1 5 7
5 6 7 2