Created
February 23, 2018 04:22
hoge
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
# frozen_string_literal: true | |
# 各行、各列の合計値の正解を計算する関数 | |
def calc_general_sum(magic_table) | |
general_sum = 0 | |
magic_table.each do |line| | |
line.include?(0) && next | |
general_sum = line.inject(:+) | |
break | |
end | |
# もし合計値がわからない場合は、行列入れ替えて再帰 | |
general_sum.zero? && general_sum = calc_general_sum(magic_table.transpose) | |
return general_sum | |
end | |
# 受け取った魔法陣を出力する関数 | |
def print_magic_table(magic_table) | |
magic_table.each do |line| | |
puts line.join(' ') | |
end | |
end | |
# 0を含む行のの配列を受け取って、正しい配列に変更 | |
def get_complete_line(line, complete_num) | |
line_str = line.join(' ').sub!(/0/, complete_num.to_s) | |
line = line_str.split(' ').map(&:to_i) | |
return line | |
end | |
# 0を含む魔法陣を受け取って、0の値を正しいものに直して返す関数 | |
def complete_magic_table(magic_table, general_sum) | |
magic_table.each_with_index do |line, i| | |
line.count(0).zero? && next | |
if line.count(0) == 1 | |
complete_num = general_sum - line.inject(:+) | |
magic_table[i] = get_complete_line(line, complete_num) | |
elsif line.count(0) == 2 | |
# 0が横に並んでいる時は、大変だから行列を入れ替えて再帰 | |
magic_table = complete_magic_table(magic_table.transpose, general_sum) | |
magic_table = magic_table.transpose | |
end | |
end | |
return magic_table | |
end | |
# 入力受取部分 | |
table_size = gets.to_i | |
input_magic_table = [] | |
table_size.times do | |
input_line = gets.split(' ').map(&:to_i) | |
input_magic_table.push(input_line) | |
end | |
general_sum = calc_general_sum(input_magic_table) | |
true_magic_table = complete_magic_table(input_magic_table, general_sum) | |
print_magic_table(true_magic_table) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment