c++ - (z-xi)^2 的最小化

Question

如果我想找到一个中位数(相当于最小化一个函数|z - x_i|)，我可以使用下面的代码片段:

std::vector<int> v{5, 6, 4, 3, 2, 6, 7, 9, 3};

std::nth_element(v.begin(), v.begin() + v.size()/2, v.end());
std::cout << "The median is " << v[v.size()/2] << '\n';

有没有这样的东西，找到“median”来最小化(z-x_i)^2？也就是说，我想找到一个数组元素，其中这些函数的总和将是最小的。

Answer 1

如果您想根据比较(z - xi) ^ 2 的谓词找到nth_element()，您可以将相应的逻辑添加到您的二元谓词中可以选择传递给 nth_element():

auto trans = [=](int xi){ return (z - xi) * (z - xi); };
std::nth_element(v.begin(), v.begin() + v.size() / 2, v.end(),
    [&](int v0, int v1) { return trans(v0) < trans(v1); });

从问题来看，z 或 xi 是否是变化的变量并不清楚。从它的外观来看，我假设 xi 是 x_i。如果 z 发生变化，只需重命名 lambda trans 中的参数(我刚刚还在捕获中提供了一个 = ...)。