最小二乘法拟合二元n次曲线C++实现

/* 最小二乘法拟合二元多次曲线 */
/* 已知一堆(X, Y)数据的集合 */
class LeastSquareMethod
{
public:
	///<summary>
    ///用最小二乘法拟合二元多次曲线
    ///例如y=ax+b
    ///其中MultiLine将返回a,b两个参数。
    ///a对应MultiLine[1]
    ///b对应MultiLine[0]
    vector<double> MultiLine(const vector<double>& arrX, const vector<double>& arrY, int dimension)
    {
        int n = dimension + 1;//dimension次方程需要求 dimension+1个 系数
		vector<vector<double>> vGuass(n, vector<double>(n + 1));//高斯矩阵 例如:y=a0+a1*x+a2*x*x

		for (int i = 0; i < n; i++)
		{
		    int j;
			for (j = 0; j < n; j++)
			{
			    vGuass[i][j] = SumArr(arrX, j + i);
			}
			vGuass[i][j] = SumArr(arrX, i, arrY, 1);
		}
		return ComputeGuass(vGuass, n);
    }
	//求数组元素的n次方的和
	double SumArr(const vector<double>& arr, int n) const
    {
        double dSum = 0;
		for (int i = 0; i < arr.size(); i++)
		{
		    if (arr[i] != 0 || n != 0)
		    {
		        dSum += pow(arr[i], n);
		    }
			else
			{
			    dSum += 1;
			}
		}
		return dSum;
    }

	double SumArr(const vector<double>& arr1, int n1, const vector<double>& arr2, int n2) const
    {
        double dSum = 0;
		for (int i = 0; i < arr1.size(); i++)
		{
		    if ((arr1[i] != 0 || n1 != 0) && (arr2[i] != 0 || n2 != 0))
		    {
		        dSum += pow(arr1[i], n1) * pow(arr2[i], n2);
		    }
			else
				dSum += 1;
		}
		return dSum;
    }

	vector<double> ComputeGuass(vector<vector<double>>& Guass, int n)
    {
        int i, j;
		int k, m;
		double temp;
		double max;
		double s;
		vector<double> x(n, 0.0);

		for (j = 0; j < n; j++)
		{
		    max = 0;
			k = j;
			for (i = j; i < n; i++)
			{
			    if (abs(Guass[i][j]) > max))
			    {
			        max = Guass[i][j];
					k = i;
			    }
			}
			if (k != j)
			{
			    for (m = j; m < n + 1; m++)
			    {
			        temp = Guass[j][m];
					Guass[j][m] = Guass[k][m];
					Guass[k][m] = temp;
			    }
			}

			if (0 == max)
			{
				// "此线性方程为奇异线性方程"
			    return x;
			}
			for (i = j + 1; i < n; i++)
			{
			    s  = Guass[i][j];
				for (m = j; m < n + 1; m++)
				{
				    Guass[i][m] = Guass[i][m] - Guass[j][m] * s / (Guass[j][j]);
				}
			}
		}

		for (i = n - 1; i >= 0; i--)
		{
		    s = 0;
			for (j = i + 1; j < n; j++)
			{
			    s = s + Guass[i][j] * x[j];
			}
			x[i] = (Guass[i][n] - s) / Guass[i][i];
		}
		return x;//返回值是函数的系数
    }
};

数学原理完全不懂,根据C#代码实现的C++版本。

  • 1
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 2
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值