[ruby-list:44361] Re: 配列のシャッフル

永井です。

>> 最初，aの要素は，5個あるので，collect は5回実行されると思うので，
 ここが問題です。実際、問題のrandomize!では、

[1,2,3,4,5].randomize!

としてもブロック内の処理は５回は実行されません。

※collectが５回というよりは、collect内の処理(yield)が配列の各要素に対して１回ずつ、
計５回実行されます。

以下のようにすると、５回実行されていないのが分かります。

--------------------------------
class Array
    def randomize!
      result = collect { p "a"; slice!(rand(length)) }
      replace result
    end

    def randomize
        arr = self.dup
        arr.randomize!
        arr
    end
end

a = [1,2,3,4,5]
p a.randomize!
--------------------------------
C:\>ruby problem.rb
"a"
"a"
"a"
[3, 4, 5]
--------------------------------


成瀬さんが貼って下さったソースを見ると、５回実行されない理由が分かります。

--------------------------------
static VALUE
rb_ary_collect(ary)
    VALUE ary;
{
    long i;
    VALUE collect;

    if (!rb_block_given_p()) {
        return rb_ary_new4(RARRAY(ary)->len, RARRAY(ary)->ptr);
    }

    collect = rb_ary_new2(RARRAY(ary)->len);
    for (i = 0; i < RARRAY(ary)->len; i++) {
        rb_ary_push(collect, rb_yield(RARRAY(ary)->ptr[i]));
    }
    return collect;
}
--------------------------------

注目すべきは for (i = 0; i < RARRAY(ary)->len; i++) { です。
RARRAY(ary)->lenはRubyで言えば、array.lengthメソッドですね。

例えば、a = [1,2,3,4,5]として、問題のrandomize!を実行したとしましょう。
すると、最初の状態で、for (i = 0; i < RARRAY(ary)->len; i++) { の部分は下記のようになります。

for (i = 0; 0 < 5; i++) {

ここで、forループが１回終わると、iとRARRAY(ary)->lenの値は次のようになります。

i                          #=> 1
RARRAY(ary)->len #=> 4

ここでRARRAY(ary)->len が５から4に減った理由は、slice!で要素が取り除かれたためです。
そして、これが問題です。

２回目のループ後には
 i                          #=> 2
RARRAY(ary)->len #=> 3

 ３回目のループ後には
 i                          #=> 3
RARRAY(ary)->len #=> 2

ここでfor (i = 0; i < RARRAY(ary)->len; i++) に値をいれてみると
for (i = 0; 3 < 2; i++) となり、ループの継続条件が「偽」になるので、４回目の処理には行かず、ループは終了します。

よって、５回実行されるはずだった処理が３回で終わってしまい、結果の配列の長さは３となってしまいます。

 いかがでしょうか？


07/12/18 に NARUSE, Yui <[e-mail]> さんは書きました:
>
> 成瀬です。
>
> Maehara Masahide (前原正英) wrote:
> >> 最初，aの要素は，5個あるので，collect は5回実行されると思うので，
> >> a.slice!(rand(a.length)) を5回繰り返してみました。
> >
> > 以下の私の実行例だと，(*)の部分以下がダウトですね。
> >
> >> ====================================================================
> >> irb(main):027:0> a = [1,2,3,4,5]
> >> => [1, 2, 3, 4, 5]
> >> irb(main):028:0> a.slice!(rand(a.length))
> >> => 4
> >> irb(main):029:0> a
> >> => [1, 2, 3, 5]
> >> irb(main):030:0> a.slice!(rand(a.length))
> >> => 5
> >> irb(main):031:0> a
> >> => [1, 2, 3]
> >> irb(main):032:0> a.slice!(rand(a.length))
> >> => 1
> >> irb(main):033:0> a
> >> => [2, 3]
> >
> > (*)
> > ここの時点で，aの長さが2になっており，collect できない！
> > 私が最初，勘違いをしたのは，collect は，しだいに減っていく a に対して
> > 新たに実行されていくものと思っていました。
> > このような理解をしたのですが，よろしいでしょうか?
>
> たぶんソースを見た方が早いので、
> 以下に array.c の該当部分のソースを挙げておきました。
>
> static VALUE
> rb_ary_collect(ary)
>     VALUE ary;
> {
>     long i;
>     VALUE collect;
>
>     if (!rb_block_given_p()) {
>         return rb_ary_new4(RARRAY(ary)->len, RARRAY(ary)->ptr);
>     }
>
>     collect = rb_ary_new2(RARRAY(ary)->len);
>     for (i = 0; i < RARRAY(ary)->len; i++) {
>         rb_ary_push(collect, rb_yield(RARRAY(ary)->ptr[i]));
>     }
>     return collect;
> }
>
>
> --
> NARUSE, Yui  <[e-mail]>
> DBDB A476 FDBD 9450 02CD 0EFC BCE3 C388 472E C1EA
>
>