#677708 (Crystal) No.757 チャンパーノウン定数 (2)

• 問題: No.757 チャンパーノウン定数 (2)
• ユーザー: te-sh
• 提出日時: 2021-07-12 16:05:44
• 言語: Crystal (1.11.2)
• 結果: AC
• 実行時間: 24 ms / 2,000 ms
• コード長: 5,204 bytes
• コンパイル時間: 20,706 ms
• コンパイル使用メモリ: 266,280 KB
• 実行使用メモリ: 6,984 KB
• 最終ジャッジ日時: 2023-09-14 20:36:15
• 合計ジャッジ時間: 23,655 ms
• ジャッジサーバーID （参考情報）: judge12 / judge13

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	3 ms; 4,928 KB
testcase_01	AC	3 ms; 4,860 KB
testcase_02	AC	3 ms; 4,916 KB
testcase_03	AC	3 ms; 4,876 KB
testcase_04	AC	3 ms; 4,828 KB
testcase_05	AC	2 ms; 4,908 KB
testcase_06	AC	3 ms; 4,840 KB
testcase_07	AC	2 ms; 4,828 KB
testcase_08	AC	3 ms; 4,928 KB
testcase_09	AC	2 ms; 4,828 KB
testcase_10	AC	3 ms; 4,884 KB
testcase_11	AC	2 ms; 4,848 KB
testcase_12	AC	3 ms; 4,940 KB
testcase_13	AC	3 ms; 4,868 KB
testcase_14	AC	2 ms; 4,876 KB
testcase_15	AC	3 ms; 4,952 KB
testcase_16	AC	3 ms; 4,900 KB
testcase_17	AC	2 ms; 4,800 KB
testcase_18	AC	3 ms; 4,968 KB
testcase_19	AC	3 ms; 4,916 KB
testcase_20	AC	3 ms; 4,916 KB
testcase_21	AC	3 ms; 4,900 KB
testcase_22	AC	3 ms; 4,888 KB
testcase_23	AC	2 ms; 4,864 KB
testcase_24	AC	3 ms; 4,860 KB
testcase_25	AC	3 ms; 4,880 KB
testcase_26	AC	3 ms; 4,812 KB
testcase_27	AC	3 ms; 4,880 KB
testcase_28	AC	14 ms; 6,476 KB
testcase_29	AC	16 ms; 6,536 KB
testcase_30	AC	3 ms; 5,168 KB
testcase_31	AC	3 ms; 5,248 KB
testcase_32	AC	3 ms; 4,964 KB
testcase_33	AC	4 ms; 5,384 KB
testcase_34	AC	7 ms; 5,840 KB
testcase_35	AC	3 ms; 5,324 KB
testcase_36	AC	3 ms; 5,156 KB
testcase_37	AC	13 ms; 6,076 KB
testcase_38	AC	13 ms; 6,520 KB
testcase_39	AC	14 ms; 6,512 KB
testcase_40	AC	20 ms; 6,460 KB
testcase_41	AC	20 ms; 6,456 KB
testcase_42	AC	6 ms; 6,276 KB
testcase_43	AC	6 ms; 6,328 KB
testcase_44	AC	24 ms; 6,984 KB
testcase_45	AC	17 ms; 6,488 KB
testcase_46	AC	6 ms; 6,224 KB
testcase_47	AC	21 ms; 6,496 KB
testcase_48	AC	12 ms; 5,952 KB
testcase_49	AC	14 ms; 6,500 KB
testcase_50	AC	3 ms; 4,844 KB
testcase_51	AC	3 ms; 4,908 KB
testcase_52	AC	3 ms; 4,892 KB
testcase_53	AC	3 ms; 4,868 KB

ソースコード

require "big"

def solve(io)
  b = io.get
  s = io.get(String)
  d = s.to_big_i(b) - 1

  io.put_e 1 if d == 1

  len = ->(x : Int32) do
    bp = b.to_big_i ** x
    bp * x - (bp - 1) // (b - 1)
  end

  f = (0..s.size).bsearch { |x| len.call(x) > d } || s.size + 1
  d -= len.call(f-1)

  g = (d // f) + (b.to_big_i ** (f - 1))
  h = d % f
  io.put g.to_s(b)[h]
end

struct Int
  def cdiv(b : Int)
    (self + b - 1) // b
  end

  def bit?(i : Int)
    bit(i) == 1
  end

  def set_bit(i : Int)
    self | (self.class.new(1) << i)
  end

  def reset_bit(i : Int)
    self & ~(self.class.new(1) << i)
  end

  {% if compare_versions(env("CRYSTAL_VERSION") || "0.0.0", "0.34.0") < 0 %}
    def bit_length : Int32
      x = self < 0 ? ~self : self

      if x.is_a?(Int::Primitive)
        Int32.new(sizeof(self) * 8 - x.leading_zeros_count)
      else
        to_s(2).size
      end
    end
  {% end %}
end

struct Int32
  SQRT_MAX = 46_340_i32

  def isqrt
    m = SQRT_MAX
    r = (1_i32..SQRT_MAX).bsearch { |i| i**2 > self }
    r.nil? ? SQRT_MAX : r - 1
  end
end

struct Int64
  SQRT_MAX = 3_037_000_499_i64

  def isqrt
    r = (1_i64..SQRT_MAX).bsearch { |i| i**2 > self }
    r.nil? ? SQRT_MAX : r - 1
  end
end

struct Number
  {% if compare_versions(env("CRYSTAL_VERSION") || "0.0.0", "1.1.0") < 0 %}
    def zero?
      self == 0
    end

    def positive?
      self > 0
    end

    def negative?
      self < 0
    end
  {% end %}

  {% if compare_versions(env("CRYSTAL_VERSION") || "0.0.0", "0.36.0") < 0 %}
    def self.additive_identity
      zero
    end

    def self.multiplicative_identity
      new(1)
    end
  {% end %}
end

class Array
  macro new_md(*args, &block)
    {% if !block %}
      {% for arg, i in args[0...-2] %}
        Array.new({{arg}}) {
      {% end %}
      Array.new({{args[-2]}}, {{args[-1]}})
      {% for arg in args[0...-2] %}
        }
      {% end %}
    {% else %}
      {% for arg, i in args %}
        Array.new({{arg}}) { |_i{{i}}|
      {% end %}
      {% for block_arg, i in block.args %}
        {{block_arg}} = _i{{i}}
      {% end %}
      {{block.body}}
      {% for arg in args %}
        }
      {% end %}
    {% end %}
  end
end

class ProconIO
  def initialize(@ins : IO = STDIN, @outs : IO = STDOUT)
    @buf = [] of String
    @index = 0
  end

  def get(k : T.class = Int32) forall T
    get_v(k)
  end

  macro define_get
    {% for i in (2..9) %}
      def get(
            {% for j in (1..i) %}
              k{{j}}{% if j < i %},{% end %}
            {% end %}
          )
        {
          {% for j in (1..i) %}
            get(k{{j}}){% if j < i %},{% end %}
          {% end %}
        }
      end
    {% end %}
  end
  define_get

  macro define_getn
    {% for i in (2..9) %}
      def get{{i}}(k : T.class = Int32) forall T
        get(
          {% for j in (1..i) %}
            k{% if j < i %}, {% end %}
          {% end %}
        )
      end
    {% end %}
  end
  define_getn

  def get_a(n : Int, k : T.class = Int32) forall T
    Array.new(n) { get_v(k) }
  end

  def get_c(n : Int, k : T.class = Int32) forall T
    get_a(n, k)
  end

  macro define_get_c
    {% for i in (2..9) %}
      def get_c(
            n : Int,
            {% for j in (1..i) %}
              k{{j}}{% if j < i %},{% end %}
            {% end %}
          )
        a = Array.new(n) do
          get(
            {% for j in (1..i) %}
              k{{j}}{% if j < i %},{% end %}
            {% end %}
          )
        end
        {
          {% for j in (1..i) %}
            a.map { |e| e[{{j-1}}] }{% if j < i %},{% end %}
          {% end %}
        }
      end
    {% end %}
  end
  define_get_c

  macro define_getn_c
    {% for i in (2..9) %}
      def get{{i}}_c(n : Int, k : T.class = Int32) forall T
        get_c(
          n,
          {% for j in (1..i) %}
            k{% if j < i %}, {% end %}
          {% end %}
        )
      end
    {% end %}
  end
  define_getn_c

  def get_m(r : Int, c : Int, k : T.class = Int32) forall T
    Array.new(r) { get_a(c, k) }
  end

  def put(*vs)
    vs.each.with_index do |v, i|
      put_v(v)
      @outs.print i < vs.size - 1 ? " " : "\n"
    end
  end

  def put_e(*vs)
    put(*vs)
    exit
  end

  def put_f(*vs)
    put(*vs)
    @outs.flush
  end


  private def get_v(k : Int32.class); get_token.to_i32; end
  private def get_v(k : Int64.class); get_token.to_i64; end
  private def get_v(k : UInt32.class); get_token.to_u32; end
  private def get_v(k : UInt64.class); get_token.to_u64; end
  private def get_v(k : Float64.class); get_token.to_f64; end
  private def get_v(k : String.class); get_token; end

  private def get_token
    if @buf.size == @index
      str = @ins.read_line
      @buf = str.split
      @index = 0
    end
    v = @buf[@index]
    @index += 1
    v
  end

  private def put_v(vs : Enumerable)
    vs.each_with_index do |v, i|
      @outs.print v
      @outs.print " " if i < vs.size - 1
    end
  end

  private def put_v(v)
    @outs.print v
  end
end

macro min_u(a, b)
  {{a}} = Math.min({{a}}, {{b}})
end

macro max_u(a, b)
  {{a}} = Math.max({{a}}, {{b}})
end

macro zip(a, *b, &block)
  {{a}}.zip({{*b}}) {{block}}
end

solve(ProconIO.new)