C++的爬山算法

news2024/10/5 18:27:19

爬山算法（Hill Climbing Algorithm）是一种局部搜索算法，它通过迭代搜索的方式寻找问题的局部最优解。在爬山过程中，算法总是选择当前状态邻域中最好（即函数值最大或最小）的状态作为下一个状态，直到达到局部最优解。由于爬山算法只关注当前状态及其邻域，因此它可能会陷入局部最优解而非全局最优解。

以下是一个用C++实现的简单爬山算法示例，用于求解一元函数的局部最大值。在这个例子中，我们将使用一个简单的二次函数 `f(x) = -x^2 + 4x - 3`，它有一个全局最大值点 `x = 2`。代码如下。

#include <iostream>
#include <cmath>
#include <random>
#include <chrono>

// 目标函数
double f(double x) {
    return -x * x + 4 * x - 3;
}

// 爬山算法
double hillClimbing(double start, double stepSize, int maxIter) {
    double current = start;
    double currentVal = f(current);
    double next, nextVal;

    for (int i = 0; i < maxIter; ++i) {
        double left = current - stepSize;
        double right = current + stepSize;
        nextVal = std::max(f(left), f(right));

        if (nextVal > currentVal) {
            // 如果邻域中存在更好的解，则移动到该解
            if (f(left) > f(right)) {
                next = left;
            } else {
                next = right;
            }
            current = next;
            currentVal = nextVal;
        } else {
            // 如果没有更好的解，则停止搜索
            break;
        }
    }

    return current;
}

int main() {
    // 设置初始点、步长和最大迭代次数
    double start = 0.0;  // 可以随机初始化或设置为特定值
    double stepSize = 0.1;
    int maxIter = 100;

    // 运行爬山算法并输出结果
    double result = hillClimbing(start, stepSize, maxIter);
    std::cout << "找到最大值 x = " << result << ", f(x) = " << f(result) << std::endl;

    return 0;
}

结果如下图所示。