最小二乘法求解线性回归模型

前言

NWAFU P1538

---

一、题目描述

假设有一组观测数据（X，Y）=（Xi , Yi) ,i=1,2,3...m，(m<20)，初步判断Y和X之间存在线性关系，如下图中左图所示。则X和Y之间的线性回归模型可表示为：y=wx+b，如果求得模型参数w和b，即可确定其数学模型。在此，基于均方误差最小化的“最小二乘法”确定其最佳模型参数，求解模型参数w和b的公式如下图中右图所示：

    请编写程序实现这一需求，其中主函数int main(void)已编写完成，需编写如下两个函数完成相应功能：

      1)  为数组的每个元素赋值，要求其函数原型:

     void input(double da[], int n);

    2)  根据输入一组观测值，求解最佳模型参数w和b（保留2位小数，当w和b绝对值小于1e-3时，则w和b取0.0），要求其函数原型为: 

     void LeastSquare(double x[], double y[], int n, double *w, double *b);

     可采用如下main(void)函数进行测试（只供参考，请勿提交）:

---

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

#define N 20
void LeastSquare(double x[], double y[], int num, double *w, double *b);
void input(double dv[], int n);

int main() {
	double x[N], y[N];
	double  w=0.0, b=0.0;
	int n;

	scanf("%d",&n);
	input(x,n);
	input(y,n);
	LeastSquare(x,y,n,&w,&b);
	printf("%.2lf %.2lf\n", w, b);
	return 0;
}

输入

观测值的数量和一组观测值X和Y，格式：
观测值数n
X1 X2 X3 ......
Y1 Y2 Y3 ......

输出

参数w和b，中间用空格分割，小数点后保留2位小数

样例输入 复制

10
5.34 2.48 8.83 3.26 1.47 4.61 6.22 6.54 5.43 8.29
10.5 6.84 15.5 8.93 6.25 11.0 11.3 12.0 9.71 13.60

样例输出 复制

1.15 4.51

二、设计步骤

套公式，代码实现如下：

void LeastSquare(double x[],double y[],int num,double *w,double *b)
{
    double w1 = 0.0,w2 = 0.0,x0 = 0.0;      //定义w表达式中的两个部分以及x的平均值
    int i;

    //求x的平均值
    for(i = 0;i < num;i++){
        x0 += x[i];
    }
    x0 /= num;
    
    //求w
    for(i = 0;i < num;i++){
        w1 += y[i]*(x[i] - x0);
        w2 += x[i]*x[i];
    }

    *w = w1 / (w2 - (num * x0 * x0));
    //求b
    for(i = 0;i < num;i++){
        *b += y[i] - (*w)*x[i];
    }
    *b /= num;
   
    //判断绝对值！读题时千万不要忽略掉！
    if(((*w) < 1e-3&&(*w) > 0)||((*w) > -1e-3&&(*w) < 0))
        (*w) = 0.00;
    if(((*b) < 1e-3&&(*b) > 0)||((*b) > -1e-3&&(*b) < 0))
        (*b) = 0.00;
            }

//数组赋值函数
void input(double dv[],int n)
{
    int i;
    for(i = 0;i < n;i++){
        scanf("%lf",&dv[i]);
    }
}

---

总结

认真读题！！！！！！！！！！！！