Question

在Matlab中，我们可以得到复数向量的最大值，最小值以及位置：

[mag_x idx_x] = max(abs(X));
[mag_y idx_y] = max(abs(Y));

我知道我们可以用C ++做类似的事情，因为Matlab是用C ++构建的。但我无法理解。我也有不完整的代码，仍然在努力：

#include <iostream>     // std::cout
#include <algorithm>    // std::min_element, std::max_element
#include <vector>
#include <complex>
#include <iostream>
int main()
{

//typedef std::complex < double > C
vector <complex<short> > vec;

vec.push_back ( complex <short> (1,1));
vec.push_back ( complex <short> (1,2));
vec.push_back ( complex <short> (3,4));
vec.push_back ( complex <short> (0,0));
vec.push_back ( complex <short> (4,8));
vec.push_back ( complex <short> (0,0));
vec.push_back ( complex <short> (2,3));
vec.push_back ( complex <short> (1,1));

int size = sizeof(x_vec)/sizeof(x_vec[0]);
for(int i =0; i < size; i++)
{
    cout << x_vec[i] << endl;
}
 ...

return 0;
}

我还会查看此链接中的示例代码：http://www.cplusplus.com/reference/algorithm/max_element/但它仅适用于矢量。

另外当我计算size = sizeof(x_vec)/sizeof(x_vec[0]);我期待大小= 8（因为上面有8个向量复合体）但我只给了我size = 6.

Answer 1

要将vector<complex>与std::max_element一起使用，您需要提供一个比较两个complex对象的值的函数。您需要这样做，因为有多种方法可以比较它们。

一种方法是编写一个lambda函数，使用complex比较两个std::abs es。您可以在致电max_element：

时写下此内容

// C++14
auto maxelem = max_element(vec.begin(), vec.end(),
                           [](auto a, auto b) { return abs(a) < abs(b); });
// C++11
auto maxelem = max_element(vec.begin(), vec.end(),
                           [](const complex<short>& a, const complex<short>& b)
                             { return abs(a) < abs(b); });

cout << "Max: " << *maxelem << endl;

max_element生成一个包含找到的最大值的迭代器。可以使用std::distance：

找到向量中迭代器的索引

cout << "Max position: " << std::distance(vec.begin(), maxelem) << endl;

complex比较器函数可以存储在auto中以供重用：

// C++14
auto absLess = [](auto a, auto b) { return abs(a) < abs(b); };
// C++11
auto csAbsLess = [](const complex<short> &a,
                    const complex<short> &b) { return abs(a) < abs(b); };

我们可以将其用于各种其他标准库算法，例如min_element和sort。

auto maxelem = max_element(vec.begin(), vec.end(), absLess);
cout << "Max: " << *maxelem << endl;
cout << "Max position: " << std::distance(vec.begin(), maxelem) << endl;

auto minelem = min_element(vec.begin(), vec.end(), absLess);
cout << "Min: " << *minelem << endl;
cout << "Min position: " << std::distance(vec.begin(), minelem) << endl;

sort(vec.begin(), vec.end(), absLess);

for (const auto &c : vec) {
  cout << c << endl;
}

输出：

Max: (4,8)
Max position: 4
Min: (0,0)
Min position: 3
(0,0)
(0,0)
(1,1)
(1,1)
(1,2)
(2,3)
(3,4)
(4,8)

当然，它不像[mag_x idx_x] = max(abs(X));那样雄辩！

An online demonstration is on Wandbox here.

如何在C ++中获得向量复数值的最大值，最小值和位置？

1 个答案: