或るプログラマの開発日記

日々の勉強したことの備忘録なんかに使っていきます

Rubyのinject/reduceメソッドに関するメモ

つい最近までRubyのinjectやらreduceが何してるか分からんかったので調査したことをメモ。

inject/reduceメソッド

injectメソッドは、ブロックを使って繰り返し計算を行うのに使います。ブロックに順に「要素1、要素2」、「ブロックの前回の戻り値、要素3」、「ブロックの前回の戻り値、要素4」、...を渡します。メソッドの戻り値はブロックが最後に返した値になります。

http://ref.xaio.jp/ruby/classes/enumerable/inject

イメージとしては、ハッシュや配列(要はEnumerableオブジェクト)の各要素に対して繰り返し処理を行い一つに纏めた結果を返す処理に使うものかなーという感じです。

因みにreduceは、injectメソッドの別名となります。mapとcollectの関係みたいなものですね。

使い方いろいろ

injectメソッドの使用例として、配列の合計を求める方法が良く使われます。
以下、サンプルとしていろんな書き方を試してみました。

array = (1..20).to_a

p array.inject {|sum, n| sum + n }#=>210
p array.inject(0){|sum, n| sum + n }#=>210
p array.inject(0,:+)#=>210
p array.inject(:+)#=>210

#100を初期値にしてみる
p array.inject(100,:+)#=>310

injectの書き方にもブロックで要素を処理する方法や、初期値を指定または省略する方法、またメソッドの引数でシンボルを渡して処理する方法があります。

また、配列の合計を計算する以外の使い方もあります。

次の例では初期値を空のハッシュで定義し、レシーバの配列から条件に合致する値のみを追加したハッシュを返す処理を行っています。初期値を配列にすれば、結果を配列で返すことも可能です。

array = (1..10).to_a.reduce({}) do |arr, item |
  arr << item if item % 2 == 0
  arr
end

p array
#=>{2=>true, 4=>true, 6=>true, 8=>true, 10=>true}

工夫次第で、配列周りの処理を簡潔に記述できそうですね。

参考URL

reduceやinjectという名称と処理のイメージが中々結び付かなかったのですが、以下のURLの記事の説明がなかなか分かりやすかったです。
Rubyist Magazine - map と collect、reduce と inject ―― 名前の違いに見る発想の違い

Rubyで多重ループを抜ける方法

Rubyの大域脱出に関するメモ。

ネストが深いループから何かの条件で一気に抜けたい場合、Rubyではcatch-throwを使うのが適切なようです。

書き方

以下が大域脱出のサンプル。如何にもサンプルの為のサンプルという感じですが、ネストしたループから抜けているのがお分かりいただけるかと。

catch(:break_loop) do
  (1..5).each do |x|
    (1..5).each do |y|
      puts "x=%d, y=%d" % [x,y]
      throw :break_loop  if x * y > 10
    end
  end
end

実行結果

x=1, y=1
x=1, y=2
x=1, y=3
x=1, y=4
x=1, y=5
x=2, y=1
x=2, y=2
x=2, y=3
x=2, y=4
x=2, y=5
x=3, y=1
x=3, y=2
x=3, y=3
x=3, y=4

JavaC++に慣れた人だと、catch節には例外処理を書くものだというイメージがあったりするのですが、Rubyの場合はその辺は切り離して考えたほうが良さそうです。

オブジェクトを返すこともできます

catch内部に定義した変数を返却する事も可能。こういう書き方を覚えておくと何かの時に役に立つのかもしれません。

(a,b,c) = catch(:break_loop) do
  (1..20).each do |x|
    (1..20).each do |y|
      (1..20).each do |z|
        throw(:break_loop, [x,y,z])  if x * y * z > 1000
      end
    end
  end
end
puts "a=%d, b=%d, c=%d" % [a,b,c]#=>a=3, b=17, c=20

雑感

RubyはGOTO文がありませんので、上の例以外に一気にネストを抜ける為には例外をraiseするか、メソッドに分割してreturnする方法ぐらいしかありませんね。

Rubyのヒアドキュメントを使ってみる

Rubyのヒアドキュメントに関するメモ。文字列を扱うちょっとしたプログラムで結構使えます。

ヒアドキュメントとは

普通の文字列リテラルはデリミタ(", ', ` など)で囲まれた 文字列ですが、ヒアドキュメントは `<<識別子' を含む行の次の行から `識別子' だけの行の直前までを文字列とする行指向のリテラルです。

https://docs.ruby-lang.org/ja/latest/doc/spec=2fliteral.html#here

ヒアドキュメントを使うメリットとして、

  • 複数行の文章を変数で扱う場合、コード上にそのまま記述可能。
  • ダブルコーテーション、シングルコーテーションを含む文字列もエスケープ無しで記述可能。

があります。

ヒアドキュメントの書き方

一口にヒアドキュメントといっても、色んな書き方があるのでまとめてみます。

基本

下の例はEOSを識別子としています。慣例的にEOSまたはEOFを使うことが多いようですが、開始と終了の識別子が合っていればなんでもかまいません。

puts <<EOS
"Hello World!"
EOS
#=>"Hello World!"
式展開

ヒアドキュメントは、通常の文字列と同じく式展開が可能です。シングルコーテーションで開始の識別子を囲むと式展開はされません。

name = "Taro"
puts <<"STR1"
Hello! #{name}
STR1
#=>Hello! Taro

puts <<'STR2'
Hello! #{name}
STR2
#=>Hello! #{name}
インデント

通常は終了の識別子を行頭に書かないとエラーになりますが、識別子の前にハイフンをつけると、終了の識別子をインデントできます。

3.times do
  puts <<-EOS
    Hello World!
  EOS
end
#=>    Hello World!
#=>    Hello World!
#=>    Hello World!

但し、上の例では余計なインデントがヒアドキュメントに含まれてしまいます。Ruby2.3以降ではチルダをつけることで、余計なインデントを除去してくれます。

3.times do
  puts <<~EOS
    Hello World!
  EOS
end
#=>Hello World!
#=>Hello World!
#=>Hello World!
コマンド実行

以下のように書くことでコマンド実行ができます。Windowsで動作確認済み。

puts <<`ENV`
env 
ENV

SQLなんかを扱うのに便利

個人的に良く使う例として、SQLのようなダブルコーテーション、シングルコーテーションがある文字列を扱うのに便利なので、複数のinsert文やselect文を生成したい時に使います。

s =<<STR1
0001
0002
0003
0004
0005
STR1
s.split.each do |str1|
  [1,2].each do|num1|
    puts <<~SQL
      select * from table where col1 = "#{str1}" and col2 = #{num1};"
    SQL
  end
end

実行結果

select * from table where col1 = "0001" and col2 = 1;"
select * from table where col1 = "0001" and col2 = 2;"
select * from table where col1 = "0002" and col2 = 1;"
select * from table where col1 = "0002" and col2 = 2;"
select * from table where col1 = "0003" and col2 = 1;"
select * from table where col1 = "0003" and col2 = 2;"
select * from table where col1 = "0004" and col2 = 1;"
select * from table where col1 = "0004" and col2 = 2;"
select * from table where col1 = "0005" and col2 = 1;"
select * from table where col1 = "0005" and col2 = 2;"

CodeIQ 「ロンリー・ルーク」問題の解答コード公開

久々の投稿になります。
今回はCodeIQの問題の解答コードを公開します。

codeiq.jp

問題

今回の問題の概要です。

  • 自然数 n, k に対し、縦横 n×n のマス目にチェスのルークの駒を k 個配置します。このとき、自身から見て上下方向・左右方向のいずれにも他の駒が存在しないような駒を「はぐれルーク」と呼びます。
  • 1 つのマスに 2 個以上の駒を置かないよう、ランダムに駒を配置します。このとき、はぐれルークの個数の期待値を F(n, k) と定義します。
  • 自然数 n, m に対し、1 ≦ k ≦ m の範囲の自然数 k に対する F(n, k) の和を G(n, m) と定義します。
  • 1000*G(n, m)の整数部分の値を出力してください。

方針

F(n, k)について、k=1~mまでの和を求めるので、F(n,k)の値の求め方を考えます。

  • k=1の場合、「はぐれルーク」は1個なので期待値は1
  • k > n-1の2乗+1の場合、どう配置しても「はぐれルーク」は無いので期待値は0
  • kの値が上記以外の場合について、全配置の組み合わせについて「はぐれルーク」の個数を計算

まあ、要はしらみつぶしというやつですww

解答

以下が、提出したコード。n×n の盤面を1次元配列で表現しているのが多少の工夫といえば工夫かも。

def func(n, k)
  return 1.0 if k == 1
  return 0.0 if ((n - 1) ** 2 ) + 1 < k
  numOfLonelyLuke = 0
  numOfCombination = 0
  (0..(n**2 - 1)).to_a.combination(k) do |array|
    numOfCombination += 1
    array.each do |x|
      isLonely = true
      array.each do |y|
        next if x == y
        if (x / n == y / n) || (x % n == y % n)
          isLonely = false
          break;
        end
      end
      numOfLonelyLuke += 1 if isLonely
    end
  end
  numOfLonelyLuke / numOfCombination.to_f
end
(n, m) = gets.split.map(&:to_i)
num = 0.0
(1..m).each do |k|
  num += func(n, k)
end
puts (num *1000).to_i

感想

今までもCodeIQの問題は解いてましたが、公開するのは今回が初めてです。
また、この機会に他の方の解答コードを見てみましたが、エレガントな解法で実装されてる方が結構いらっしゃったので驚きでした。
次はもう少し綺麗な解き方を目指すようにしないとなー。

ま、自分もまだまだ精進が必要だなーと感じ入りました。ほんと。(´Д`)

2017年5月の目標

どうも、パイソンです。

 

本日から5月。先月に引き続き月間の目標でも立ててみようかなーと思います。

 

とはいえ、先月は毎日更新を目指すとか息巻いてみたものの結局挫折したんで、今月は普通に達成出来る感じのものにしようかと。(^^;

 

というわけで、立ててみた5月の目標。

 

1.rubyでなんか作る

先月来からrubyの勉強を進めているが、やはりcuiのプログラムだけ作ってても退屈なもんです。

 

なので、今月はrubyを使って少し規模の大きなソフトを書いてみようかと思います。何作るかは検討中ですが、早い内に決めて、ブログに作成過程なんかを残して行こうかと思います。

 

2.記事更新は10記事程度を目処に

1カ月ブログやってきて、さすがに毎日更新はしんどいと実感したので、今月は少し控えめな目標にしてみましたw

 

とはいえ、数値目標を立てて置かないと更新しないまま放置になるので、一応少なすぎるという事は無い程度の規模感で。

 

 

というわけで、5月はマイペースで更新していきます。

 

ブログ開始から1カ月経過しました

どうも、パイソンです。

 

ブログ開始からちょうど1カ月経過しましたので、これまでの振り返りをしてみようと思います。

 

月初の目標

一応、今月の初めに2点ほど目標を立てておりました。

1.毎日更新を目指す事

2.rubyの勉強を頑張る事。

 

んで、1の目標ですが、結局毎日更新はできませんでした。(´・ω・`)

 

理由としては、飲み会があって帰りが遅くなった日があったり、そもそもネタが全然浮かばなかったりというのがありますが、まあ、これらのことは言訳にしかなりません。(^^;

 

つか、毎日更新とかキツイわ。本当、毎日更新してるブロガーのひとは凄いと思います。

 

次に、2の目標。

 

rubyのほうは、ブログ続けてきた効果か、大分理解が深まったかなーと個人的には思います。やはり、わからない事を調べつつ、その内容をブログにアウトプットしていくことで、学んだ事が頭に定着しているという実感はあります。

 

とはいえ、まだまだ基本的な部分しか理解出来てないという自覚があるので、来月以降も継続して勉強は続けていこうかと思います。

 

まとめ

というわけで、

1.毎日更新は無理ゲーだった

2.今後も勉強した事はブログに残して行こう

 

以上が今月の感想です。

5月もブログは継続していきます。

 

 

Rubyにはインクリメント演算子(++)が無い

Rubyを書いてて、いつも疑問に思ってた事項なので自分用にメモ。

CやJava、またはPerl等のプログラム言語では、整数の値に1を加算する時にインクリメント演算子++を使用します。しかし、Rubyではインクリメント演算子を使うことは出来ません。

インクリメント演算子が無い理由

Rubyの開発者であるMatzさんが、過去この件について言及されておりました。

3) 記号的な記法

これは単なる私の趣味ですが, 単項インクリメントとかがたまに欲しく
なります. i += 1 でいいわけですが. i++ と書いて怒られる (^^;


すんません.この件は以前から指摘されているのですが(演算子はC
に似ているのに++と--は対応する演算子が無い),++の動作が本質
的に「変数を操作する」ものであるため,変数がオブジェクトでな
Rubyでは導入できないでいます.++や--の「オブジェクト指向
意味」がRubyの他の部分と整合性を保ったまま定義できれば採用し
たいのですが….

http://blade.nagaokaut.ac.jp/cgi-bin/scat.rb/ruby/ruby-list/5323

以下、自分なりの解釈。

Rubyでは整数値はオブジェクトであるためインクリメント演算子を実現しようとすれば変数が参照しているオブジェクトそのものに作用する必要がありますが、++演算子が変数に対して作用するものであるため実現が困難であるということですかねぇ。

また、RubyのFixnumクラスはimmutableですので、さらに実現は困難ですね。

Ruby の Fixnum クラスは immutable です。 つまり、オブジェクト自体を破壊的に変更することはできません。 Bignum も同様です。

https://docs.ruby-lang.org/ja/latest/class/Fixnum.html

インクリメントメソッド

尚、Fixnumクラスにはnextsuccという値に+1するメソッドはあります。しかし、前述の通りFixnum自体がimmutableである為、

num=10
num.next

としてもnumの値は変化しません。

結論

Rubyでは、整数をインクリメントする場合はおとなしく+=1を使いましょう。