During hacking on various tasks, I needed to partition a set of elements quite a few times. I have attacked the problem with different homegrown implementations, mostly involving *select*-ing every element belonging into the same basket in turn. Fortunately I run across *divide* recently, which does exactly this… No more wheel reinvention! Let’s see a concrete example.

I have an input file like this:

a 53 2 3 b 8 62 1 23 a 9 0 31 b 4 45 4 16 7 b 1 23 c 3 42 2 31 4 6 a 1 3 22 a 7 83 1 23 3 b 1 14 4 15 16 2 c 5 16 2 34

The goal is to sum up all the numbers in rows beginning with the same character (e.g. to sum up all the numbers that are in a row beginning with ‘a’). The result should look like:

[{"a"=>241}, {"b"=>246}, {"c"=>145}]

This is an ideal task for divide! Let’s see one possible solution for the problem:

```
require 'set'
input = Set.new open('input.txt').readlines.map{|e| e.chomp}
groups = input.divide {|x,y| x.map[0][0] == y.map[0][0] }
#build the array of hashes
p groups.map.inject([]) {|a,g|
#build the hashes for the number sequences with same letters
a << g.map.inject(Hash.new(0)) {|h,v|
#for every sequence, sum the numbers it contains
h[v[0..0]] += v[2..-1].split(' ').inject(0) {|c,x|
c+=x.to_i; c}; h
}; a
}
```

The output is:

```
[{"a"=>241}, {"b"=>246}, {"c"=>145}]
```

Great - it works! Now let's take a look into the code...

The 3rd line loads the lines into a set like this:

The real thing happens on line 4. After it's execution, *groups* looks like:

, , }>

As you can see, the set is correctly partitioned now - with almost no effort! We did not even need to require an external library...

The rest of the code is out of the scope of this article (everybody is always complaining about the long articles here, so I am trying to keep them short) - and anyway, the remaining snippet is just a bunch of calls to inject. If inject does not feel too natural to you, don't worry - it took me months until I got used to it, and some people (despite of the fact that they fully understand and are able to use it) never reach after it - I guess it's a matter of taste...'

What happens if you have 2 lines with the same numbers – or does that never happen?

Ben,

theoretically that could happen – there is absolutely no guarantee that two rows will be different.

Well… it seems even

divideis not perfect (it could do the partitioning into a Hash, where the value set would be the number of occurrences of the sets or something) – or maybe this is possible somehow?Going to check it…

So why complicate it with divide?

Why not:

#

h = {}

open(‘input.txt’).each

line{|l|i; c}}h[l[0..0]] +=

l[2..-1].split(â€˜ â€˜).inject(0) {|c,x| c+=x.to

}

p h.to_a.map{|k,v| {k => v}

#

?

It would also remove the same-line weakness.

Well, the original idea was to show how Set.divide works… Admiring my just found shiny new hammer, I guess this problem pretty much looked like a big nail 🙂

Thanks for the solution, Aur.

Come on, you can do better than that 🙂

There is a method Set.classify, related to Set.divide, which gets you closer to your goal:

require ‘set’

input = Set.new open(‘input.txt’).readlines.map{|line| line.chomp.split}

groups = input.classify {|e| e.shift}

result = groups.map {|k,v| {k => v.to

a.flatten.inject(0) {|sum, x| sum+x.toi}}}