KevDog
KevDog

Reputation: 5803

Find all subsets of size N in an array using Ruby

Given an array ['a', 'b', 'c', 'd', 'e', 'f'], how would I get a list of all subsets containing two, three, and four elements?

I'm quite new to Ruby (moving from C#) and am not sure what the 'Ruby Way' would be.

Upvotes: 6

Views: 3535

Answers (2)

tokland
tokland

Reputation: 67880

Tweaking basicxman's a little bit:

2.upto(4).flat_map { |n| array.combination(n).to_a }
#=> [["a", "b"], ["a", "c"], ["a", "d"], ..., ["c", "d", "e", "f"]]

Upvotes: 7

basicxman
basicxman

Reputation: 2115

Check out Array#combination

Then something like this:

2.upto(4) { |n| array.combination(n) }

Upvotes: 14

Related Questions