大家好我有以下问题:
INPUT :
2
ababaa
AA
输出:
11个
3个
解释
对于第一种情况,字符串的后缀是“ababaa”,“babaa”,“abaa”,“baa”,“aa”和“a”。每个字符串与字符串“ababaa”的相似性分别为6,0,3,0,1,1。因此答案是6 + 0 + 3 + 0 + 1 + 1 = 11.
对于第二种情况,答案是2 + 1 = 3.
这部分有效,但我的代码应该通过的一些测试没有。
这是我的代码
def input_data
#STDIN.flush
tries = gets.chomp
end
strings=[];
tries = input_data until (tries =~ /^[1-9]$/)
tries = tries.to_i
strings << input_data until (strings.count == tries)
strings.map do |x|
values = 0
current = x.chars.to_a
(0..x.length-1).map do |possition|
current_suffix = x[possition..-1].chars.to_a
(0..current_suffix.length-1).map do |number|
if (current_suffix[0] != current[0])
break
end
if ( current_suffix[number] == current[number] )
values = values+1
end
end
end
if (values != 0)
puts values
end
end
有任何建议如何解决?
答案 0 :(得分:9)
gets
会返回nil
,但不能chomp
。因此,您需要确保在调用chomp
之前处理实际字符串。 ruby中的一个常见习惯是使用||=
运算符来设置变量,只有它是nil。所以你会写:
tries = gets # get input
tries ||= '' # set to empty string if nil
tries.chomp! # remove trailing newline