結果

問題 No.1785 Inequality Signs
ユーザー maguroflymagurofly
提出日時 2021-12-14 10:00:39
言語 Ruby
(3.3.0)
結果
AC  
実行時間 398 ms / 2,000 ms
コード長 5,038 bytes
コンパイル時間 35 ms
コンパイル使用メモリ 11,764 KB
実行使用メモリ 22,148 KB
最終ジャッジ日時 2023-09-30 17:58:04
合計ジャッジ時間 14,824 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge15
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 94 ms
15,528 KB
testcase_01 AC 146 ms
16,624 KB
testcase_02 AC 310 ms
20,312 KB
testcase_03 AC 386 ms
21,436 KB
testcase_04 AC 98 ms
15,348 KB
testcase_05 AC 218 ms
18,108 KB
testcase_06 AC 354 ms
21,316 KB
testcase_07 AC 103 ms
15,624 KB
testcase_08 AC 344 ms
20,632 KB
testcase_09 AC 141 ms
16,712 KB
testcase_10 AC 275 ms
19,220 KB
testcase_11 AC 151 ms
17,196 KB
testcase_12 AC 196 ms
17,980 KB
testcase_13 AC 136 ms
16,592 KB
testcase_14 AC 233 ms
18,304 KB
testcase_15 AC 210 ms
17,904 KB
testcase_16 AC 180 ms
17,216 KB
testcase_17 AC 278 ms
19,320 KB
testcase_18 AC 279 ms
19,624 KB
testcase_19 AC 240 ms
18,360 KB
testcase_20 AC 261 ms
19,208 KB
testcase_21 AC 245 ms
18,624 KB
testcase_22 AC 144 ms
16,876 KB
testcase_23 AC 178 ms
17,684 KB
testcase_24 AC 383 ms
21,476 KB
testcase_25 AC 208 ms
18,036 KB
testcase_26 AC 202 ms
17,816 KB
testcase_27 AC 391 ms
22,148 KB
testcase_28 AC 398 ms
21,908 KB
testcase_29 AC 391 ms
21,780 KB
testcase_30 AC 390 ms
21,808 KB
testcase_31 AC 393 ms
21,904 KB
testcase_32 AC 394 ms
21,836 KB
testcase_33 AC 92 ms
15,288 KB
testcase_34 AC 104 ms
15,652 KB
testcase_35 AC 116 ms
16,576 KB
testcase_36 AC 291 ms
20,740 KB
testcase_37 AC 246 ms
21,728 KB
testcase_38 AC 226 ms
20,840 KB
testcase_39 AC 142 ms
17,044 KB
testcase_40 AC 168 ms
17,800 KB
testcase_41 AC 207 ms
19,144 KB
testcase_42 AC 325 ms
21,908 KB
testcase_43 AC 372 ms
21,812 KB
testcase_44 AC 328 ms
20,540 KB
testcase_45 AC 157 ms
17,900 KB
testcase_46 AC 228 ms
20,556 KB
testcase_47 AC 177 ms
18,768 KB
testcase_48 AC 221 ms
19,368 KB
testcase_49 AC 99 ms
15,644 KB
testcase_50 AC 323 ms
20,572 KB
testcase_51 AC 114 ms
16,096 KB
testcase_52 AC 107 ms
15,760 KB
testcase_53 AC 126 ms
16,136 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Syntax OK

ソースコード

diff # 

main = -> {
    N, K = gets.split.map(&:to_i)
    
    # table[i] = (K+i)C(N)
    # table[i] = table[i - 1] * (K + i) / (K - N + i)
    table = [K.mod_nCr(N)]
    (0 ... N - K).each do |i|
        table[i] = 0
    end
    table[N - K] = 1 if N - K >= 0
    ([N - K + 1, 1].max .. N).each do |i|
        table[i] = table[i - 1] * (K + i).mod_div(K - N + i) % MOD
    end
    
    f = Factorial.new(N)
    
    ans = 0
    (0 .. N).each do |le|
        # p [le, f.comb(N - 1, le), table[le]]
        ans += f.comb(N - 1, le) * table[le]
        ans %= MOD
    end
    
    puts ans
}

DEBUG = true
MOD = 10**9+7
YESNO = %w(No Yes)
INF = 10**9

require "set"
require "prime"

def int; gets.to_s.to_i end
def ints; gets.to_s.split.map { |s| s.to_i } end
def int1s; gets.to_s.split.map { |s| s.to_i - 1 } end
def float; gets.to_s.to_f end
def floats; gets.to_s.split.map { |s| s.to_f } end
def array_of(&convert); gets.to_s.split.map(&convert) end
def string; gets.to_s.chomp end
def rep(n, &b); Array.new(n, &b) end
def yes; puts YESNO[1] end
def no; puts YESNO[0] end
def yesno t; puts YESNO[t] end
def zip(xs, *yss); Enumerator.new { |y| xs.zip(*yss) { |a| y.yield(*a) } } end
def max(*xs, &block); block_given? ? xs.max_by(&block) : xs.max end
def min(*xs, &block); block_given? ? xs.min_by(&block) : xs.min end
def minmax(*xs, &block); block_given? ? xs.minmax_by(&block) : xs.minmax end
def gcd(*xs); xs.inject(0, :gcd) end
def matrix(h, w, fill=nil, &block); return Array.new(h) { Array.new(w, &block) } if block_given?; Array.new(h) { [fill] * w } end
def debug(x = nil); STDERR.puts (block_given? ? yield(x) : x).inspect if DEBUG; x end
def debug_grid(grid, width = 1); grid.each { |row| STDERR.puts row.map { |x| x.inspect.ljust(width) }.join("") } if DEBUG; grid end
def if_debug; yield if DEBUG end

module Boolean
  def coerce(other); [other, to_i] end
  def +@; to_i end
  def to_int; to_i end
  def *(other); to_i * other end
end

class TrueClass
  include Boolean
  def to_i; 1 end
end

class FalseClass
  include Boolean
  def to_i; 0 end
end

class Integer
  def div_ceil(y); (self + y - 1) / y end
  def mod_inv(mod = MOD); pow(mod-2, mod) end
  def mod_div(y, mod = MOD); self * y.mod_inv(mod) % mod end
  def mod_nCr(r, mod = MOD); x = y = 1; (1..r).each { |i| x = x * (self + 1 - i) % mod; y = y * i % mod }; x.mod_div(y); end
  def factorial(mod = MOD); (2..self).inject(1) { |f, x| f * x % mod } end
  def popcount; x = self; c = 0; while x > 0; c += 1 if x & 1 == 1; x >>= 1 end; c end #TODO: faster
  def bitbrute(&block); (1<<self).times(&block) end
  def nCr(r); x = 1; (1..r).each { |i| x *= self + 1 - i; x /= i }; x; end
  def each_divisor; return Enumerator.new { |y| each_divisor { |k| y << k } } unless block_given?; k = 1; while k * k < self; if self % k == 0; yield k; yield self / k end; k += 1; end; yield k if k * k == self end
  def divisors; each_divisor.to_a end
end

class Range
  def end_open; exclude_end? ? self.end : self.end + 1 end
  def end_close; exclude_end? ? self.end - 1 : self.end end
  def upper_bound; ac, wa = self.begin, self.end_open; while wa - ac > 1; if yield((wj = (ac + wa) / 2)); ac = wj else wa = wj end; end; yield(ac) ? ac : nil end
  def lower_bound; ac, wa = self.end_open, self.begin; while ac - wa > 1; if yield((wj = (ac + wa) / 2)); ac = wj else wa = wj end; end; yield(ac) ? ac : nil end
  def shakutori(r2, &pred); Enumerator.new { |y| j, r = r2.begin, r2.end_open; each { |i| j += 1 while j + 1 < r and pred[i, j+1]; y.yield(i, j) } }; end
  def widest(&block); Enumerator.new { |y| j, n = self.begin, self.end_open; each { |i| j += 1 while j < n and block[i, j]; y.yield(i, j) if block[i, j] } } end
end

class Array
  def power(&block); (0 ... 1 << size).each(&block) end
  def sorted_merge(other); a = []; i = j = 0; n, m = size, other.size; if j < m and other[j] < self[i]; a << other[j]; j += 1 else; a << self[i]; i += 1 end while i < n; a.push(*other[j..-1]) if j < m; a end
  def upper_bound; ac, wa = 0, size; while wa - ac > 1; if yield(self[(wj = (ac + wa) / 2)]); ac = wj else; wa = wj end; end; ac end
  def lower_bound; ac, wa = size, 0; while wa - ac > 1; if yield(self[(wj = (ac + wa) / 2)]); ac = wj else; wa = wj end; end; ac end
  def cum(*xs, &op); a = []; a << xs[0] if xs.size > 0; a << x = self[0]; (1...size).each { |i| a << x = op[x, self[i]] }; a end
  def cumdiff(range); self[range.end_open] - self[range.begin]; end
end

module Enumerable
  def sorted_uniq; x = nil; filter { |y| c = x === y; x = y; !c } end
  def cumsum; ys = [0]; each { |x| ys << x + ys[-1] }; ys end
end

class Factorial
  def initialize(max, mod = MOD)
    @mod = mod
    @fac = [1, 1]
    @fin = [1, 1]
    @inv = [nil, 1]
    (2 .. max).each do |i|
      @fac[i] = @fac[i - 1] * i % mod
      @inv[i] = mod - @inv[mod % i] * (mod / i) % mod
      @fin[i] = @fin[i - 1] * @inv[i] % mod
    end
  end
  
  def fact(n)
    @fac[n]
  end
  
  def comb(n, k)
    return 0 if n < k or n < 0 or k < 0
    @fac[n] * @fin[k] % @mod * @fin[n - k] % @mod
  end
end

main[]
0