您的功能实现中有一个错误。
atan(b*x - 1 / 2)
该术语1 / 2
进行整数除法并计算为 0,这可能不是您想要的。通常,在对双变量进行算术运算时使用双字面量。该pow()
函数确实(double, int)
作为它的重载之一,所以你在那里很好。它也有一个(double, double)
重载,但如果你的指数实际上是一个整数,那么你不想要那个。
这是最简单的求根方法的简单实现——二分法(后来我注意到OP使用二分标记,完美)。
#include <iostream>
#include <cmath>
#include <random>
double f(const double x, const double a, const double b, const double c)
{
return sin((a*x / (1.0 + pow(x, 2))) + 1.0) * atan(b*x - 1.0 / 2.0) + exp(-c*x) * atan(x);
}
double BisectionMethod(
double f(double, double, double, double),
const double a, const double b, const double c,
const std::random_device::result_type entropy)
{
std::mt19937 gen(entropy);
static const auto lower_bound = -1.0;
static const auto upper_bound = 1.0;
std::uniform_real_distribution<> dis(lower_bound, upper_bound);
auto pos_pt = dis(gen);
auto neg_pt = dis(gen);
while (f(pos_pt, a, b, c) < 0.0)
pos_pt = dis(gen);
while (f(neg_pt, a, b, c) > 0.0)
neg_pt = dis(gen);
static const auto about_zero_mag = 1E-8;
for (;;)
{
const auto mid_pt = (pos_pt + neg_pt)/2.0;
const auto f_mid_pt = f(mid_pt, a, b, c);
if (fabs(f_mid_pt) < about_zero_mag)
return mid_pt;
if (f_mid_pt >= 0.0)
pos_pt = mid_pt;
else
neg_pt = mid_pt;
}
}
int main()
{
double a, b, c;
std::random_device rd;
static const auto entropy = rd();
a =10, b = 2.0, c = 0.0;
const auto root1 = BisectionMethod(f, a, b, c, entropy);
std::cout << "a = " << a << ", b = " << b << ", c = " << c << std::endl;
std::cout << "Found root: (" << root1 << ", " << f(root1, a, b, c) << ")" << std::endl;
a =4.5, b = 2.8, c = 1.0;
const auto root2 = BisectionMethod(f, a, b, c, entropy);
std::cout << "a = " << a << ", b = " << b << ", c = " << c << std::endl;
std::cout << "Found root: (" << root2 << ", " << f(root2, a, b, c) << ")" << std::endl;
}
输出:
g++ -O3 -std=c++11 -Wall -Wextra -pedantic main.cpp -o root && ./root
a = 10, b = 2, c = 0
Found root: (0.143042, -2.12425e-09)
a = 4.5, b = 2.8, c = 1
Found root: (0.136172, 5.81247e-09)
每次运行时输出都会发生变化,因为这使用了 RNG。从视觉上看,输出看起来正确。
该代码假定根以 -1.0 和 1.0 为界,这在您的情况下是正确的。如果您希望它更通用,那么您需要添加用于处理溢出和检查 nans 的逻辑。如果根不在 -1.0 和 1.0 之间,这将永远循环。尽管如此,它解决了这个问题中的具体问题,并且是更普遍的事情的开始。
另请注意,您的函数有多个根,并且给定的代码只找到一个根。
编辑:清理代码。添加entropy
作为参数,BisectionMethod()
以便它是可重现的,如果我们正在谈论数值方法,这似乎是可取的。