牛客题解 | 实现求解线性方程组的共轭梯度法

题目

题目链接

共轭梯度法是一种求解线性方程组的迭代方法。具体步骤如下：

1. 计算初始残差向量$r$
   $$r=b-Ax$$
2. 计算初始搜索方向向量$p$
   $$p=r$$
3. 迭代更新$x$和$r$和$p$，直到满足收敛条件
   $$x=x+\alpha p$$
   r i + 1 = r i − α ( A p i ) r_{i+1} = r_i - \alpha (A p_i)