在Mathematica中实现Newton-Raphson迭代的收敛时间算法(一般三次多项式)

考察一般的三次多项式,以r为参数:

复制代码
p[z_, r_] := z^3 + (r - 1) z - r;
roots[r_] := z /. Solve[p[z, r] == 0, z];

此多项式的根为:

尽管看起来这个多项式是特殊的,其实一般的三次多项式都是可以通过线性变换化为这个形式的。此时的Newton-Raphson映射为:

此时,可以编写带有参数r的收敛速度计数函数:

复制代码
NewNewtonCounter = 
  Compile[{{z, _Complex}, {r, _Real}, {otherroot, _Complex}},
   Module[{counter = 0, zold = N[z] + 1, znew = N[z]},
    If[Abs[znew] < 10^(-9), znew = 10^(-9) + 0.0 I,
     znew = znew];
    For[counter = 0,
     (Abs[zold - znew] > 10^(-6)) && (counter < 85), counter++,
     (zold = znew; znew = (r + 2*zold^3)/(-1 + r + 3*zold^2))];
    Which[Abs[znew - 1] < 10^(-4), counter,
     Abs[znew - otherroot] < 10^(-4), 85 + counter,
     Abs[znew - Conjugate[otherroot]] < 10^(-4), 170 + counter,
     True, 255
     ]
    ]
   ];
 
   (*返回给定区域中的每个点相应的收敛计数*)  
 NewNewtonArray[r_, {{remin_, remax_}, {immin_, immax_}}, steps_] :=
 Module[{croot = -N[(1 + Sqrt[1 - 4 r])/2]},
  Table[NewNewtonCounter[x + y I, r, croot],
   {y, immin, immax, (immax - immin)/steps},
   {x, remin, remax, (remax - remin)/steps}
   ]
  ]

region = NewNewtonArray[2, {{-2, 2}, {-2, 2}}, 1000];
NewtonPlot1[{{-2, 2}, {-2, 2}}, region, NewtonColorRGB]
复制代码
region = NewNewtonArray[0.05, {{-2, 2}, {-2, 2}}, 1000];
NewtonPlot1[{{-2, 2}, {-2, 2}}, region, NewtonColorRGB]
相关推荐
0+1113 分钟前
算法 --栈
算法·leetcode
鶴哥只手遮天6 分钟前
Gmsh 源码深度解析-第 2 章 几何内核
算法·开源
鶴哥只手遮天12 分钟前
Gmsh 源码深度解析-第 9 章 后处理与插件体系
算法·开源
鶴哥只手遮天15 分钟前
Gmsh 源码深度解析-第 7 章 网格质量、优化与高级技术
算法·开源
鶴哥只手遮天20 分钟前
OpenFOAM 源码深度剖析 · 第二季-15 实战:手写一个求解器、库与边界条件
算法·开源
鶴哥只手遮天22 分钟前
Gmsh 源码深度解析-第 8 章 数值内核与高阶有限元
算法·开源
鶴哥只手遮天24 分钟前
Gmsh 源码深度解析-第 4 章 网格生成总控与尺寸场
算法·开源
鶴哥只手遮天25 分钟前
Gmsh 源码深度解析-第 10 章 求解器框架、API 与语言绑定
算法·开源
鶴哥只手遮天30 分钟前
OpenFOAM 源码深度剖析 · 第二季-02 燃烧与化学反应:`combustionModel`、化学求解器与 Arrhenius 反应率
算法·开源
鶴哥只手遮天30 分钟前
Gmsh 源码深度解析-第 6 章 三维体网格算法
算法·开源