娃娃鸭学Ruby-35、编写自定义的迭代器

自定义的迭代器

可调用一个与方法调用相关联的代码块

def twice
yield
yield
end

def sequence(n,m,c)
i=0
while(i<n)
yield m*i +c
i+=1
end
end

sequence(3,5,1){|y| puts y}
irb(main):008:0> sequence(3,5,1){|y| puts y}
1
6
11
=> nil

更复杂代码，
迭代器内部使用另一迭代器
def circle(r,n)
n.times do |i|
angle=Math::PI * 2*i /n
yield r*Math.cos(angle),r*Math.sin(angle)
end
end

def sequence(n,m,c)
i,s=0,[]
while(i<n)
y=m*i+c
yield y if block_given?
s<<y
i+=1
end
s
end

2011-4-19 14:59 danny