結果

問題 No.1518 Simple Combinatorics
ユーザー yuruhiyayuruhiya
提出日時 2021-05-28 21:02:28
言語 Crystal
(1.10.0)
結果
AC  
実行時間 3 ms / 2,000 ms
コード長 2,291 bytes
コンパイル時間 19,004 ms
コンパイル使用メモリ 254,820 KB
実行使用メモリ 4,528 KB
最終ジャッジ日時 2023-08-07 05:05:45
合計ジャッジ時間 19,143 ms
ジャッジサーバーID
(参考情報)
judge15 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,376 KB
testcase_01 AC 3 ms
4,416 KB
testcase_02 AC 2 ms
4,384 KB
testcase_03 AC 3 ms
4,380 KB
testcase_04 AC 2 ms
4,444 KB
testcase_05 AC 2 ms
4,500 KB
testcase_06 AC 2 ms
4,380 KB
testcase_07 AC 3 ms
4,392 KB
testcase_08 AC 2 ms
4,528 KB
testcase_09 AC 3 ms
4,448 KB
testcase_10 AC 2 ms
4,484 KB
testcase_11 AC 3 ms
4,380 KB
testcase_12 AC 2 ms
4,376 KB
testcase_13 AC 2 ms
4,380 KB
testcase_14 AC 3 ms
4,484 KB
testcase_15 AC 2 ms
4,528 KB
testcase_16 AC 2 ms
4,416 KB
testcase_17 AC 3 ms
4,380 KB
testcase_18 AC 3 ms
4,460 KB
testcase_19 AC 2 ms
4,484 KB
testcase_20 AC 2 ms
4,416 KB
testcase_21 AC 2 ms
4,384 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

# require "math/Mint"
macro static_modint(name, mod)
  struct {{name}}
    MOD = {{mod}}.to_i64

    def self.zero
      new
    end

    def self.raw(value : Int64)
      result = new
      result.value = value
      result
    end

    @value : Int64

    def initialize
      @value = 0i64
    end

    def initialize(value)
      @value = value.to_i64 % MOD
    end

    def initialize(m : self)
      @value = m.value
    end

    protected def value=(value : Int64)
      @value = value
    end

    getter value : Int64

    def + : self
      self
    end

    def - : self
      self.class.raw(value != 0 ? MOD &- value : 0i64)
    end

    def +(v)
      self + v.to_m
    end

    def +(m : self)
      x = value &+ m.value
      x &-= MOD if x >= MOD
      self.class.raw(x)
    end

    def -(v)
      self - v.to_m
    end

    def -(m : self)
      x = value &- m.value
      x &+= MOD if x < 0
      self.class.raw(x)
    end

    def *(v)
      self * v.to_m
    end

    def *(m : self)
      self.class.new(value &* m.value)
    end

    def /(v)
      self / v.to_m
    end

    def /(m : self)
      raise DivisionByZeroError.new if m.value == 0
      a, b, u, v = m.to_i64, MOD, 1i64, 0i64
      while b != 0
        t = a // b
        a &-= t &* b
        a, b = b, a
        u &-= t &* v
        u, v = v, u
      end
      self.class.new(value &* u)
    end

    def //(v)
      self / v
    end

    def **(exponent : Int)
      t, res = self, self.class.raw(1i64)
      while exponent > 0
        res *= t if exponent & 1 == 1
        t *= t
        exponent >>= 1
      end
      res
    end

    def ==(v)
      value == v
    end

    def ==(m : self)
      value == m.value
    end

    def succ
      self.class.raw(value != MOD &- 1 ? value &+ 1 : 0i64)
    end

    def pred
      self.class.raw(value != 0 ? value &- 1 : MOD &- 1)
    end

    def abs
      self
    end

    def to_i64 : Int64
      value
    end

    delegate to_s, to: @value
    delegate inspect, to: @value
  end

  struct Int
    def to_m : {{name}}
      {{name}}.new(self)
    end
  end

  class String
    def to_m : {{name}}
      {{name}}.new(self)
    end
  end
end

static_modint(Mint, 10**9 + 7)
n, k = read_line.split.map(&.to_i)
n = n.to_m
puts n**(k + 1) - n * (n - 1)**k
0