数学建模社区-数学中国

标题: Matlab中自定义的函数怎么能支持向量化操作? [打印本页]

作者: 红薯 时间: 2013-9-26 18:28
标题: Matlab中自定义的函数怎么能支持向量化操作?

function r=primeQ(n)
r=1;
for i=2:sqrt(n);
if mod(n,i)==0;
r=0;
return
end
end;

复制代码

上面的primeQ是个判断素数的函数，调用的时候只会这样用
sum( arrayfun(@primeQ, 2:1e6) )
听说arraygfun的速度不快，我希望可以支持向量化操作，就像下面这样
sum(primeQ(2:1e5))
应该怎么做？
注意：我知道判断素数有更好的算法，我关心的不是算法问题

作者: lvyanlong 时间: 2013-9-27 12:56
你可以使用arrayfun这个函数

你首先测试是否为偶数，如果不是，后边的偶数因子可以全部跳过，这样可以减少更多的时间
当然可以也对3做同样的操作，判断数不是2或3的倍数后只要判断那些6k+1和6k+5的因子
可以继续下去，但是越往后提升的效果就越差，所以自行斟酌

对于比较大的数可以考虑用Fermat小定理或者Miller-Rabin检验去判断；双精度最大连续整数上限是2^53

另外可以用matlab coder把m文件转换成mex，这样会更快

欢迎光临数学建模社区-数学中国 (http://www.madio.net/)