お題：文字列を先頭から見て同じところまで除去

ブログに久しぶりにGroovyのコードが載った気がする。

複数の文字列を受け取り、受け取った文字列をそれぞれ先頭から見てゆき、
すべてが同じ内容であれば除去した内容の文字列を返却する関数を書いて下さい。
※関数の引数と戻り値については複数の文字列が受け渡しできれば型や方法は
問いません。
例１）hoge("abcdef", "abc123")
戻り値 => "def" と "123" ("abc" が除去される）
例２）hoge("あいうえお", "あいさんさん", "あいどる")
戻り値 => "うえお", "さんさん", "どる" ("あい" が除去される）
例３）hoge("12345", "67890", "12abc")
戻り値 => "12345", "67890", "12abc" (一致なしのため、そのまま返却される）

テストコード

// お題
assert hoge('abcdef', 'abc123') == ['def', '123']
assert hoge("あいうえお", "あいさんさん", "あいどる") == ["うえお", "さんさん", "どる"]
assert hoge("12345", "67890", "12abc") == ["12345", "67890", "12abc" ]


//@irof の テスト
assert hoge('123', '123') == ['','']
assert hoge('112233', '123123') == ['12233', '23123']
assert hoge('なまむぎ', 'なまごめ', 'なまたまご') == ['むぎ', 'ごめ', 'たまご']
assert hoge('abcde', 'abcdefg') == ['', 'fg']
assert hoge('abcdefg', 'abcde') == ['fg', ''] // 追加したテスト

解答

def hoge(String... args) {
    work = args.toList()
    work.sort()[0..-2].each {
        for (i in 0 .. args*.size().min()-1 ) {
            m =  ( it[0 .. i] =~ work[-1][0 .. i++] )
            if(m.find()) {
                del = m[0]
            }
        }   
    }
    args*.replaceFirst(~/$del/,"")
}

// お題のテスト
assert hoge('abcdef', 'abc123') == ['def', '123']
assert hoge("あいうえお", "あいさんさん", "あいどる") == ["うえお", "さんさん", "どる"]
assert hoge("12345", "67890", "12abc") == ["12345", "67890", "12abc" ]

//@irof の テスト
assert hoge('123', '123') == ['','']
assert hoge('112233', '123123') == ['12233', '23123']
assert hoge('なまむぎ', 'なまごめ', 'なまたまご') == ['むぎ', 'ごめ', 'たまご']
assert hoge('abcde', 'abcdefg') == ['', 'fg']
assert hoge('abcdefg', 'abcde') == ['fg', ''] // 追加したテスト

メモ

1行目 def hoge(String... args)

Java SE 5の可変引数で導入されてた可変引数(可変長引数)も、もちろん使える。

Groovy物語 Story #8 メソッド？クロージャ？でも、少し触れられてる。

2行目 work = args.toList()

toList():配列をListに変換プログラミングGROOVY 184

3行目 work.sort()[0..-2].each {

sort():collectionの各要素をソートプログラミングGROOVY p87
[0..-2]
.. :繰り返しの範囲の指定プログラミングGROOVY p81
-2 :末尾から2番目の要素プログラミングGROOVY p82

4行目 for (i in 0 .. args*.size().min()-1 )

for文: in で指定もできる。プログラミングGROOVY p108
args*.:展開ドットコレクションの全ての要素に対して適用できるプログラミングGROOVY p94
size():Javaの場合は、length()、size()とバラバラだが、Groovyはsize()で統一されている。プログラミングGROOVY p182
min(): Conllectionの要素のうち最小のものを抽出プログラミングGROOVY p86

5行目 m = ( it[0 .. i] =~ work[-1][0 .. i++] )

=~: 正規表現、Matcherオブジェクトの生成プログラミングGROOVY p114
[-1]:末尾の要素プログラミングGROOVY p82

6行目 if(m.find())

Matcherオブジェクトの評価プログラミングGROOVY p115

7行目 del = m[0]

マッチした要素の取り出し。Javaの正規表現 APIの機能。
正規表現にマッチした部分を取り出す

11行目 replaceFirst(~/$del/,"")

$del:GString $変数名の展開プログラミングGROOVY p67
replaceFirst():文字列の置換 replaceAll()での説明がプログラミングGROOVY p117
replaceAll()は全置換、replaceFirst()は最初にマッチしたものだけを置換

15行目 assert

プログラミングGROOVY p32,p111
このような間違ったコードを書く

def hoge(String... args) {
    
    args.sort()[0..-2].each {
        for (i in 0 .. args*.size().min()-1 ) {
            m =  ( it[0 .. i] =~ args[-1][0 .. i++] )
            if(m.find()) {
                del = m[0]
            }
        }   
    }
    args*.replaceFirst(~/$del/,"")
}

すると、以下のようにassert文で失敗する。

Exception thrown
2011/08/15 19:24:34 org.codehaus.groovy.runtime.StackTraceUtils sanitize

警告: Sanitizing stacktrace:

Assertion failed: 

assert hoge('abcdef', 'abc123') == ['def', '123']
       |                        |
       [123, def]               false


	at org.codehaus.groovy.runtime.InvokerHelper.assertFailed(InvokerHelper.java:386)

sort()をしてるから、要素の順番が入れ替わっていてNGになっていることがすぐ分かる。
GroovyのPower Assertは、原因の調査がしやすい。