【算法11】梯度下降法查找抛物线极值

用到了一个随机数宏RADN_MAX

/*************************************************************************
        > File Name: gradient.c
        > Author: jby
        > Mail: 
        > Created Time: Thu 20 Jul 2023 03:34:14 PM CST
 ************************************************************************/

#include<stdio.h>
#include<stdlib.h>
#include<time.h>

double gradientDesc(double a, double b, double c, int cnt) {
    double l = 0.1;
    double x = rand() / (double)RAND_MAX * 20 - 10;//RAND_MAX 宏：随机数的最大值
    printf("x = %lf\n", x);
    int i;
    for (i = 0; i < cnt; i++) {
        double gradient = 2 * a * x + b;
        if (a > 0) {
            x = x - gradient * l;
        } else {
            x = x + gradient * l;
        }
        double fx = a * x * x + b * x + c;
        printf("i = %d, x = %lf, fx = %lf, gradient = %lf\n", i, x, fx, gradient);
    }
    return x;
}


int main () {
    srand(time(0));
    int a = rand() % 20 - 10;
    int b = rand() % 20 - 10;
    int c = rand() % 20 - 10;
    
    a = a == 0 ? 1 : a;
    printf("f(x) = %dx^2 + %dx + %d\n", a, b, c);

    gradientDesc(a, b, c, 100);
    
    return 0;
}