matlab - 简化查找阿姆斯特朗数的代码

Question

A armstrong number is a number that is the sum of its own digits each raised to the power of the number of digits.

我的代码如下所示，用于查找7位阿姆斯特朗数，使用bsxfun的原因是因为它速度相当快，但它只接受三个参数(Mathematica的类似函数Outer可以接受多个参数)。如何使我的代码更加紧凑？

tic
m1=(1:9).^7;
m=(0:9).^7;
r1=bsxfun(@plus, m1', m);
r2=bsxfun(@plus, r1, reshape(m,1,1,[]));
r3=bsxfun(@plus, r2, reshape(m,1,1,1,[]));
r4=bsxfun(@plus, r3, reshape(m,1,1,1,1,[]));
r5=bsxfun(@plus, r4, reshape(m,1,1,1,1,1,[]));
r6=bsxfun(@plus, r5, reshape(m,1,1,1,1,1,1,[]));
r7= permute(r6, 7:-1:1);
A=reshape((1e6:1e7-1), size(r7));
A(A==r7)
toc

(*
ans =

     1741725
     4210818
     9800817
     9926315
*)

Answer 1

您可以使用 shiftdim 代替 reshape 并将 bsxfun 放入 for 循环中

m=(0:9).^7.';
r=(1:9).^7.';
for k=1:6,
    m = shiftdim(m,-1);
    r = bsxfun(@plus, r, m);
end
r= permute(r, 7:-1:1);
A=reshape((1e6:1e7-1), size(r7));
A(A==r7)