結果

問題 No.129 お年玉(2)
ユーザー kohei2019kohei2019
提出日時 2021-07-28 21:16:00
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 257 ms / 5,000 ms
コード長 6,710 bytes
コンパイル時間 279 ms
コンパイル使用メモリ 87,216 KB
実行使用メモリ 82,600 KB
最終ジャッジ日時 2023-10-11 19:03:19
合計ジャッジ時間 11,255 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 156 ms
79,508 KB
testcase_01 AC 257 ms
82,240 KB
testcase_02 AC 76 ms
71,860 KB
testcase_03 AC 76 ms
72,032 KB
testcase_04 AC 76 ms
71,796 KB
testcase_05 AC 206 ms
81,496 KB
testcase_06 AC 189 ms
80,352 KB
testcase_07 AC 233 ms
81,716 KB
testcase_08 AC 210 ms
81,392 KB
testcase_09 AC 232 ms
81,556 KB
testcase_10 AC 205 ms
81,492 KB
testcase_11 AC 193 ms
80,976 KB
testcase_12 AC 244 ms
82,404 KB
testcase_13 AC 202 ms
80,768 KB
testcase_14 AC 215 ms
81,540 KB
testcase_15 AC 185 ms
81,448 KB
testcase_16 AC 181 ms
81,224 KB
testcase_17 AC 250 ms
81,808 KB
testcase_18 AC 190 ms
81,092 KB
testcase_19 AC 240 ms
81,940 KB
testcase_20 AC 199 ms
81,112 KB
testcase_21 AC 211 ms
81,364 KB
testcase_22 AC 181 ms
81,320 KB
testcase_23 AC 203 ms
81,308 KB
testcase_24 AC 255 ms
81,816 KB
testcase_25 AC 188 ms
80,816 KB
testcase_26 AC 190 ms
80,952 KB
testcase_27 AC 191 ms
80,700 KB
testcase_28 AC 234 ms
81,736 KB
testcase_29 AC 75 ms
71,420 KB
testcase_30 AC 73 ms
71,892 KB
testcase_31 AC 74 ms
71,840 KB
testcase_32 AC 77 ms
71,504 KB
testcase_33 AC 119 ms
78,580 KB
testcase_34 AC 97 ms
76,972 KB
testcase_35 AC 118 ms
78,568 KB
testcase_36 AC 125 ms
78,708 KB
testcase_37 AC 121 ms
78,372 KB
testcase_38 AC 171 ms
80,348 KB
testcase_39 AC 162 ms
79,992 KB
testcase_40 AC 192 ms
81,132 KB
testcase_41 AC 189 ms
80,264 KB
testcase_42 AC 175 ms
80,364 KB
testcase_43 AC 164 ms
79,908 KB
testcase_44 AC 239 ms
82,600 KB
testcase_45 AC 171 ms
79,908 KB
testcase_46 AC 197 ms
81,084 KB
testcase_47 AC 213 ms
81,424 KB
testcase_48 AC 199 ms
81,784 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

import math
import sys
sys.setrecursionlimit(10**7)
#競技プログラミング対整数問題のライブラリーです
class segki_pro_mod():
    def __init__(self, N, ls, mod):
        self.mod = mod
        self.default = 1
        self.func = (lambda x, y: (x * y) % self.mod)
        self.N = N
        self.K = (N - 1).bit_length()
        self.N2 = 1 << self.K
        self.dat = [self.default] * (2**(self.K + 1))
        for i in range(self.N):  # 葉の構築
            self.dat[self.N2 + i] = ls[i]
        self.build()

    def build(self):
        for j in range(self.N2 - 1, -1, -1):
            self.dat[j] = self.func(self.dat[j << 1], self.dat[j << 1 | 1])  # 親が持つ条件

    def leafvalue(self, x):  # リストのx番目の値
        return self.dat[x + self.N2]

    def update(self, x, y):  # index(x)をyに変更
        i = x + self.N2
        self.dat[i] = y
        while i > 0:  # 親の値を変更
            i >>= 1
            self.dat[i] = self.func(self.dat[i << 1], self.dat[i << 1 | 1])
        return

    def query(self, L, R):  # [L,R)の区間取得
        L += self.N2
        R += self.N2
        vL = self.default
        vR = self.default
        while L < R:
            if L & 1:
                vL = self.func(vL, self.dat[L])
                L += 1
            if R & 1:
                R -= 1
                vR = self.func(self.dat[R], vR)
            L >>= 1
            R >>= 1
        return self.func(vL, vR)

class integerlib():
    def __init__(self):
        pass
    
    def primeset(self,N): #N以下の素数をsetで求める.エラトステネスの篩O(√Nlog(N))
        lsx = [1]*(N+1)
        for i in range(2,int(-(-N**0.5//1))+1):
            if lsx[i] == 1:
                for j in range(i,N//i+1):
                    lsx[j*i] = 0
        setprime = set()
        for i in range(2,N+1):
            if lsx[i] == 1:
                setprime.add(i)
        return setprime
    
    def defprime(self,N):#素数かどうかの判定、エラトステネスの篩O(√Nlog(N))
        return N in self.primeset(N)
    
    def gcd(self,ls):#最大公約数
        ls = list(ls)
        ans = 0
        for i in ls:
            ans = math.gcd(ans,i)
        return ans

    def lmc(self,ls):#最小公倍数
        ls = list(ls)
        ans = self.gcd(ls)
        for i in ls:
            ans = self.lmcsub(ans,i)
        return ans
    
    def lmcsub(self,a,b):
        gcd = math.gcd(a,b)
        lmc = (a*b)//gcd
        return lmc

    
    def factorization(self,N):#素因数分解√N
        arr = []
        temp = N
        for i in range(2, int(-(-N**0.5//1))+1):
            if temp%i==0:
                cnt=0
                while temp%i==0:
                    cnt+=1
                    temp //= i
                arr.append([i, cnt])
        if temp!=1:
            arr.append([temp, 1])
        if arr==[]:
            arr.append([N, 1])
        return arr #[素因数、個数]

    def factorizationset(self,N):#素因数分解√N,含まれている素因数の種類
        if N == 1:
            return set()
        ls = self.factorization(N)
        setf = set()
        for j in ls:
            setf.add(j[0])
        return setf

    def divisorsnum(self,N):#約数の個数
        ls = []
        for i in self.factorization(N):
            ls.append(i[1])
        d = 1
        for i in ls:
            d *= i+1
        return d

    def Eulerfunc(self,N):#オイラー関数正の整数Nが与えられる。1,2,…,Nのうち、Nと互いに素であるものの個数を求めよ。
        ls = list(self.factorizationset(N))
        ls2 = [N]
        for i in ls:
            ls2.append(ls2[-1]-ls2[-1]//i)
        return ls2[-1]

    def make_divisors(self,N):#約数列挙O(√N)
        lower_divisors , upper_divisors = [], []
        i = 1
        while i*i <= N:
            if N % i == 0:
                lower_divisors.append(i)
                if i != N // i:
                    upper_divisors.append(N//i)
            i += 1
        return lower_divisors + upper_divisors[::-1]

    def invmod(self,a,mod):#mod逆元
        a %= mod
        if a == 0:
            return 0
        if a == 1:
            return 1
        return (-self.invmod(mod % a, mod) * (mod // a)) % mod
        
    def cmbmod(self,n, r, mod):#nCr % mod
        inv = [0,1]
        for i in range(2, n + 1):
            inv.append((-inv[mod % i] * (mod // i)) % mod)
        cmd = 1
        for i in range(1,min(r,n-r)+1):
            cmd = (cmd*(n-i+1)*inv[i])%mod
        return cmd

    def permmod(self,n, r, mod):#nPr % mod
        perm = 1
        for i in range(n,r-1,-1):
            perm = (perm*i)%mod
        return perm

    def modPow(self,a,n,mod):#繰り返し二乗法 a**n % mod
        if n==0:
            return 1
        if n==1:
            return a%mod
        if n % 2 == 1:
            return (a*self.modPow(a,n-1,mod)) % mod
        t = self.modPow(a,n//2,mod)
        return (t*t)%mod

    def invmodls(self,n,mod):#nまでのinvmod
        inv = [0,1]
        for i in range(2, n + 1):
            inv.append((-inv[mod % i] * (mod // i)) % mod)
        return inv
    
    def factorization_all_n(self,n):#n以下の自然数すべてをを素因数分解
        lspn = [[] for i in range(n+1)]
        lsnum = [i for i in range(n+1)]
        lsp = list(self.primeset(n))
        lsp.sort()
        for p in lsp:
            for j in range(1,n//p+1):
                cnt = 0
                while lsnum[p*j]%p==0:
                    lsnum[p*j] //= p
                    cnt += 1
                lspn[j*p].append((p,cnt))
        return lspn

    def cmbmodls(self,n,mod):#二項係数逆元使えないver
        lsans = [1]
        lsp = list(self.primeset(n))
        lsp.sort()
        invp = [0]*(n+1)
        lspmod = []
        for i in range(len(lsp)):
            invp[lsp[i]] =i
            lspmod.append(lsp[i]%mod)
        lsX = [0]*(len(lsp))
        SG = segki_pro_mod(len(lsp),[1]*len(lsp),mod)
        lspn = self.factorization_all_n(n)
        for i in range(1,n+1):
            l = n-i+1
            r = i
            change = set()
            for p,cnt in lspn[l]:
                lsX[invp[p]] += cnt
                change.add(invp[p])
            for p,cnt in lspn[r]:
                lsX[invp[p]] -= cnt
                change.add(invp[p])
            changels = list(change)
            for j in changels:
                SG.update(j,self.modPow(lsp[j],lsX[j],mod))
            lsans.append(SG.dat[1])
        return lsans

N = int(input())
M = int(input())
rm = (N%(M*1000))//1000
mod = 10**9
IT = integerlib()
ls = IT.cmbmodls(M,mod)
print(ls[rm])
0