結果

問題 No.2858 Make a Palindrome
ユーザー tomerun
提出日時 2024-08-25 15:27:56
言語 Crystal
(1.14.0)
結果
AC  
実行時間 577 ms / 3,000 ms
コード長 1,895 bytes
コンパイル時間 11,815 ms
コンパイル使用メモリ 296,952 KB
実行使用メモリ 36,092 KB
最終ジャッジ日時 2024-08-25 15:28:14
合計ジャッジ時間 17,035 ms
ジャッジサーバーID
(参考情報)
judge3 / judge2
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 40
権限があれば一括ダウンロードができます

ソースコード

diff #

read_line.to_i.times do
  puts solve()
end

def solve
  n, m = read_line.split.map(&.to_i)
  s = read_line.chars
  r = manacher(s)
  max = 0
  (n * 2).times do |i|
    if i % 2 == 1
      len = (r[i] + 1) // 2
      max = {max, len * 2 - 1}.max
    else
      len = r[i] // 2
      max = {max, len * 2}.max
    end
  end
  return 1 if m <= max

  r = manacher(s + s)
  max = 0
  rep = false
  (n * 4).times do |i|
    next if !(i - r[i] <= 2 * n && 2 * n <= i + r[i])
    if i % 2 == 1
      len = (r[i] + 1) // 2
      if len * 2 - 1 > max
        max = len * 2 - 1
        rem = s[..(i // 2 - len)]
        if rem == rem.reverse
          rep = true
        end
      end
    else
      len = r[i] // 2
      if len * 2 > max
        max = len * 2
        rem = s[..(i // 2 - 1 - len)]
        if rem == rem.reverse
          rep = true
        end
      end
    end
  end
  # puts [r, max, rep]
  return 2 if m <= max

  r = manacher(s + s + s)
  max3 = 0
  (n * 6).times do |i|
    if i % 2 == 1
      len = (r[i] + 1) // 2
      max3 = {max3, len * 2 - 1}.max
    else
      len = r[i] // 2
      max3 = {max3, len * 2}.max
    end
  end

  if max3 == max + n
    (m - max + n - 1) // n + 2
  else
    -1
  end
end

def manacher(s)
  str = ['$']
  s.each do |c|
    str << c << '$'
  end
  i = 0
  j = 0
  rad = Array.new(str.size, 0)
  while i < str.size
    while i - j >= 0 && i + j < str.size && str[i - j] == str[i + j]
      j += 1
    end
    rad[i] = j
    k = 1
    while k <= i && k + rad[i - k] < j
      rad[i + k] = rad[i - k]
      k += 1
    end
    i += k
    j -= k
  end
  rad

  # pos = 1.step(to: str.size - 1, by: 2).max_by { |i| rad[i] }
  # len = (rad[pos] + 1) // 2
  # odd_max = {(pos - rad[pos] + 1) // 2, len * 2 - 1}

  # pos = 0.step(to: str.size - 1, by: 2).max_by { |i| rad[i] }
  # len = rad[pos] // 2
  # even_max = {(pos - rad[pos] + 1) // 2, len * 2}
end
0