diff --git a/README.md b/README.md index 2bd8f9d7..305c8595 100644 --- a/README.md +++ b/README.md @@ -582,3 +582,4 @@ Profile on LeetCode: [fartem](https://leetcode.com/fartem/). | 456. 132 Pattern | [Link](https://leetcode.com/problems/132-pattern/) | [Link](./lib/medium/456_132_pattern.rb) | [Link](./test/medium/test_456_132_pattern.rb) | | 462. Minimum Moves to Equal Array Elements II | [Link](https://leetcode.com/problems/minimum-moves-to-equal-array-elements-ii/) | [Link](./lib/medium/462_minimum_moves_to_equal_array_elements_ii.rb) | [Link](./test/medium/test_462_minimum_moves_to_equal_array_elements_ii.rb) | | 468. Validate IP Address | [Link](https://leetcode.com/problems/validate-ip-address/) | [Link](./lib/medium/468_validate_ip_address.rb) | [Link](./test/medium/test_468_validate_ip_address.rb) | +| 474. Ones and Zeroes | [Link](https://leetcode.com/problems/ones-and-zeroes/) | [Link](./lib/medium/474_ones_and_zeroes.rb) | [Link](./test/medium/test_474_ones_and_zeroes.rb) | diff --git a/leetcode-ruby.gemspec b/leetcode-ruby.gemspec index b33f633c..ea9cfc32 100644 --- a/leetcode-ruby.gemspec +++ b/leetcode-ruby.gemspec @@ -5,7 +5,7 @@ require 'English' ::Gem::Specification.new do |s| s.required_ruby_version = '>= 3.0' s.name = 'leetcode-ruby' - s.version = '7.1.6' + s.version = '7.1.7' s.license = 'MIT' s.files = ::Dir['lib/**/*.rb'] + %w[README.md] s.executable = 'leetcode-ruby' diff --git a/lib/medium/474_ones_and_zeroes.rb b/lib/medium/474_ones_and_zeroes.rb new file mode 100644 index 00000000..d54d0f7b --- /dev/null +++ b/lib/medium/474_ones_and_zeroes.rb @@ -0,0 +1,29 @@ +# frozen_string_literal: true + +# https://leetcode.com/problems/ones-and-zeroes/ +# @param {String[]} strs +# @param {Integer} m +# @param {Integer} n +# @return {Integer} +def find_max_form(strs, m, n) + result = ::Array.new(m + 1) { ::Array.new(n + 1, 0) } + strs.each do |str| + zeros = 0 + ones = 0 + str.each_char do |char| + if char == '0' + zeros += 1 + else + ones += 1 + end + end + + (zeros..m).reverse_each do |i| + (ones..n).reverse_each do |j| + result[i][j] = [result[i][j], result[i - zeros][j - ones] + 1].max + end + end + end + + result[m][n] +end diff --git a/test/medium/test_474_ones_and_zeroes.rb b/test/medium/test_474_ones_and_zeroes.rb new file mode 100644 index 00000000..cd64c4fc --- /dev/null +++ b/test/medium/test_474_ones_and_zeroes.rb @@ -0,0 +1,29 @@ +# frozen_string_literal: true + +require_relative '../test_helper' +require_relative '../../lib/medium/474_ones_and_zeroes' +require 'minitest/autorun' + +class OnesAndZerosTest < ::Minitest::Test + def test_default_one + assert_equal( + 4, + find_max_form( + %w[10 0001 111001 1 0], + 5, + 3 + ) + ) + end + + def test_default_two + assert_equal( + 2, + find_max_form( + %w[10 0 1], + 1, + 1 + ) + ) + end +end