C ++查找函数的根

时间:2015-02-07 07:41:29

标签: c++ numerical-methods bisection

我被要求找到以下功能的根

sin((a*x / (1 + pow(x, 2))) + 1) * atan(b*x - 1 / 2) + exp(-c*x) * atan(x)

有两组值

  • a=10b=2c=0
  • a=4.5b=2.8c=1

但我没有获得 start end 值,我需要在其中找到根目录。我该怎么办?

注意:atan()表示tan的反函数。

代码段:

double f(double x, double a, double b, double c)
{
    return sin((a*x / (1 + pow(x, 2))) + 1) * atan(b*x - 1 / 2) + exp(-c*x) * atan(x);
}

double RootFinder(double f(double, double, double, double), double a, double b, double c, double left, double right, double precision)
{
    double f_left = f(left, a, b, c), now = left + precision, f_right = f(now, a, b, c);
    while (f_left * f_right > 0 && now < right)
    {
        f_left = f_right;
        now += precision;
        f_right = now;
    }
    return now - precision / 2;
}

1 个答案:

答案 0 :(得分:1)

您的功能实施存在一个错误。

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;
}

Output

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。视觉上输出看起来正确。

代码假设root被-1.0和1.0限制,在你的情况下是正确的。如果您希望它更通用,那么您需要添加逻辑来处理溢出和检查nans。如果根不在-1.0和1.0之间,这将永远循环。尽管如此,它解决了这个问题中的具体问题,并且是更普遍的开始。

另请注意,您的函数具有多个根,并且给定的代码只能找到一个根。

修改:清理代码。添加了entropy作为BisectionMethod()的参数,以便它可以重现,如果我们谈论数值方法,这似乎是可取的。