結果
問題 |
No.2406 Difference of Coordinate Squared
|
ユーザー |
|
提出日時 | 2025-06-24 15:52:31 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 236 ms / 2,000 ms |
コード長 | 2,852 bytes |
コンパイル時間 | 428 ms |
コンパイル使用メモリ | 82,032 KB |
実行使用メモリ | 109,236 KB |
最終ジャッジ日時 | 2025-06-24 15:52:45 |
合計ジャッジ時間 | 13,343 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 55 |
ソースコード
import os,sys,random,threading #sys.exit() 退出程序 #sys.setrecursionlimit(10**6) #调整栈空间 from random import randint,choice,shuffle #randint(a,b)从[a,b]范围随机选择一个数 #choice(seq)seq可以是一个列表,元组或字符串,从seq中随机选取一个元素 #shuffle(x)将一个可变的序列x中的元素打乱 from copy import deepcopy from io import BytesIO,IOBase from types import GeneratorType from functools import lru_cache,reduce #reduce(op,迭代对象) from bisect import bisect_left,bisect_right #bisect_left(x) 大于等于x的第一个下标 #bisect_right(x) 大于x的第一个下标 from collections import Counter,defaultdict,deque from itertools import accumulate,combinations,permutations #accumulate(a)用a序列生成一个累积迭代器,一般list化前面放个[0]做前缀和用 #combinations(a,k)a序列选k个 组合迭代器 #permutations(a,k)a序列选k个 排列迭代器 from heapq import heapify,heappop,heappush #heapify将列表转为堆 from typing import Generic,Iterable,Iterator,TypeVar,Union,List from string import ascii_lowercase,ascii_uppercase,digits #小写字母,大写字母,十进制数字 from math import ceil,floor,sqrt,pi,factorial,gcd,log,log10,log2,inf #ceil向上取整,floor向下取整 ,sqrt开方 ,factorial阶乘 from decimal import Decimal,getcontext #Decimal(s) 实例化Decimal对象,一般使用字符串 #getcontext().prec=100 修改精度 from sys import stdin, stdout, setrecursionlimit input = lambda: sys.stdin.readline().rstrip("\r\n") MI = lambda :map(int,input().split()) li = lambda :list(MI()) ii = lambda :int(input()) mod = int(1e9 + 7) #998244353 inf = 1<<60 py = lambda :print("YES") pn = lambda :print("NO") DIRS = [(0, 1), (1, 0), (0, -1), (-1, 0)] # 右下左上 DIRS8 = [(0, 1), (1, 1), (1, 0), (1, -1), (0, -1), (-1, -1), (-1, 0),(-1, 1)] # →↘↓↙←↖↑↗ mod=998244353 MI=lambda:map(int,input().split()) n,m=MI() # precompute factorials and inverse factorials fac=[1]*(n+1) for i in range(1,n+1):fac[i]=fac[i-1]*i%mod ifac=[1]*(n+1) ifac[n]=pow(fac[n],mod-2,mod) for i in range(n,0,-1):ifac[i-1]=ifac[i]*i%mod inv4=pow(4,mod-2,mod) inv4n=pow(inv4,n,mod) ans=0 if m==0: if n%2==0: c=fac[n]*ifac[n//2]%mod*ifac[n-n//2]%mod ans=(2*c*pow(2,n,mod)-c*c)%mod else: am=abs(m) d=[] i=1 while i*i<=am: if am%i==0: d.append(i) if i*i!=am:d.append(am//i) i+=1 for dd in d: for p in (dd,-dd): if m%p:continue q=m//p if (p-n)%2 or (q-n)%2:continue k1=(n-p)//2 k2=(n-q)//2 if 0<=k1<=n and 0<=k2<=n: c1=fac[n]*ifac[k1]%mod*ifac[n-k1]%mod c2=fac[n]*ifac[k2]%mod*ifac[n-k2]%mod ans=(ans+c1*c2)%mod ans=ans*inv4n%mod print(ans)