結果

問題 No.1200 お菓子配り-3
ユーザー uni_python
提出日時 2020-08-28 22:22:20
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 360 ms / 4,000 ms
コード長 3,551 bytes
コンパイル時間 242 ms
コンパイル使用メモリ 82,296 KB
実行使用メモリ 80,336 KB
最終ジャッジ日時 2024-11-20 22:35:23
合計ジャッジ時間 6,041 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

import sys
input=sys.stdin.readline
def I(): return int(input())
def MI(): return map(int, input().split())
def LI(): return list(map(int, input().split()))

def main():
    mod=10**9+7
    S=I()
    def gcd(a, b):
        while b: a, b = b, a % b
        return a
    def isPrimeMR(n):
        d = n - 1
        d = d // (d & -d)
        L = [2]
        for a in L:
            t = d
            y = pow(a, t, n)
            if y == 1: continue
            while y != n - 1:
                y = (y * y) % n
                if y == 1 or t == n - 1: return 0
                t <<= 1
        return 1
    def findFactorRho(n):
        m = 1 << n.bit_length() // 8
        for c in range(1, 99):
            f = lambda x: (x * x + c) % n
            y, r, q, g = 2, 1, 1, 1
            while g == 1:
                x = y
                for i in range(r):
                    y = f(y)
                k = 0
                while k < r and g == 1:
                    ys = y
                    for i in range(min(m, r - k)):
                        y = f(y)
                        q = q * abs(x - y) % n
                    g = gcd(q, n)
                    k += m
                r <<= 1
            if g == n:
                g = 1
                while g == 1:
                    ys = f(ys)
                    g = gcd(abs(x - ys), n)
            if g < n:
                if isPrimeMR(g): return g
                elif isPrimeMR(n // g): return n // g
                return findFactorRho(g)
    #[(p1,n1),(p2,n2),...]の形で返す
    def primeFactor(n):
        i = 2
        ret = {}
        rhoFlg = 0
        while i*i <= n:
            k = 0
            while n % i == 0:
                n //= i
                k += 1
            if k: ret[i] = k
            i += 1 + i % 2
            if i == 101 and n >= 2 ** 20:
                while n > 1:
                    if isPrimeMR(n):
                        ret[n], n = 1, 1
                    else:
                        rhoFlg = 1
                        j = findFactorRho(n)
                        k = 0
                        while n % j == 0:
                            n //= j
                            k += 1
                        ret[j] = k

        if n > 1: ret[n] = 1
        if rhoFlg: ret = {x: ret[x] for x in sorted(ret)}
        return ret


    #約数列挙
    #10^6以下で約数の数が最大なのは720720の時(240個しかない！)
    def divisors(N):
        pf = primeFactor(N)
        ret = [1]
        for p in pf:
            ret_prev = ret
            ret = []
            for i in range(pf[p]+1):
                for r in ret_prev:
                    ret.append(r * (p ** i))
        return sorted(ret)
    
    
    def calc(X,Y):
        diff=X-Y
        if diff==0:
            ans=0
            #A=1の場合，B+C=X=Y
            ans+=X-1
            
            #それ以外ならB*(A+1)=X=Y
            d=divisors(X)
            ans+=len(d)-1#(A+1)の部分がdの要素になれば良い，(A+1)は1になり得ない
            if X%2==0:# (A+1)=2 => A=1 をダブルカウントしている
                ans-=1
            return ans
        
        d=divisors(diff)
        ans=0
        for i in range(len(d)):
            A=d[i]+1
            Z=d[-1-i]
            if Y-Z>0:
                if (Y-Z)%(A+1)==0:
                    ans+=1
        return ans
        
    
    # X-Y=(B-C)*(A-1)
    for _ in range(S):
        x,y=MI()
        if y>x:
            x,y=y,x
        print(calc(x,y))
            
        


main()
0