红薯 发表于 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,这样会更快
页: [1]
查看完整版本: Matlab中自定义的函数怎么能支持向量化操作?