在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]
相关推荐
心中有国也有家1 天前
cann-recipes-infer:昇腾 NPU 推理的“菜谱集合”
经验分享·笔记·学习·算法
绝知此事1 天前
【算法突围 01】线性结构与哈希表:后端开发的收纳术
java·数据结构·算法·面试·jdk·散列表
碧海银沙音频科技研究院1 天前
通话AEC与语音识别AEC的软硬回采链路
深度学习·算法·语音识别
csdn_aspnet1 天前
Python 算法快闪 LeetCode 编号 70 - 爬楼梯
python·算法·leetcode·职场和发展
m0_629494731 天前
LeetCode 热题 100-----26.环形链表 II
数据结构·算法·leetcode·链表
壹号用户1 天前
用队列实现栈
数据结构·算法
做人求其滴1 天前
面试经典 150 题 380 274
c++·算法·面试·职场和发展·力扣
daad7771 天前
记一组无人机IMU传感器数据
算法
计算机安禾1 天前
【c++面向对象编程】第42篇:模板特化与偏特化:为特定类型定制实现
开发语言·c++·算法