在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]
相关推荐
枫景Maple27 分钟前
LeetCode 2297. 跳跃游戏 VIII(中等)
算法·leetcode
鑫鑫向栄27 分钟前
[蓝桥杯]修改数组
数据结构·c++·算法·蓝桥杯·动态规划
鑫鑫向栄28 分钟前
[蓝桥杯]带分数
数据结构·c++·算法·职场和发展·蓝桥杯
小wanga1 小时前
【递归、搜索与回溯】专题三 穷举vs暴搜vs回溯vs剪枝
c++·算法·机器学习·剪枝
天宫风子1 小时前
线性代数小述(一)
线性代数·算法·矩阵·抽象代数
hjyowl2 小时前
题解:AT_abc407_c [ABC407C] Security 2
c语言·开发语言·算法
ai产品老杨3 小时前
减少交通拥堵、提高效率、改善交通安全的智慧交通开源了。
前端·vue.js·算法·ecmascript·音视频
小于不是小鱼呀3 小时前
手撕 K-Means
人工智能·算法·机器学习
m0_740154673 小时前
K-Means颜色变卦和渐变色
算法·机器学习·kmeans