*hWSSKJr SSKJrJrJrJrJrJrJ r SSK J r SSK J r SSKJrJrJrJr SSKJr SSKJr SS KJrJr SS KJr SS KJr SS KJr SS K J!r! SSK"J#r# SSK$J%r%J&r& SSK'J(r( Sr)Sr*Sr+Sr,S9Sjr-Sr.Sr/S:Sjr0\14Sjr2Sr3Sr4Sr5Sr6Sr7Sr8S r9S:S!jr:S;S"jr;\"S#S$S%S&9S'5r<\"S(S$S%S&9S)5r=\"S*S$S%S&9S+5r>SS0jrBS 1. a and n should be relatively prime Returns ======= int : the order of ``a`` modulo ``n`` Raises ====== ValueError If `n \le 1` or `\gcd(a, n) \neq 1`. If ``a`` or ``n`` is not an integer. Examples ======== >>> from sympy.ntheory import n_order >>> n_order(3, 7) 6 >>> n_order(4, 7) 3 See Also ======== is_primitive_root We say that ``a`` is a primitive root of ``n`` when the order of ``a`` modulo ``n`` equals ``totient(n)`` r%n should be an integer greater than 1*The two numbers should be relatively prime)r ValueErrorrritemspowrint) ana_orderpepepe_orderfactorsorderpxexxs Z/home/mande/repo/quber/.venv/lib/python3.13/site-packages/sympy/ntheory/residue_ntheory.pyn_orderr3s X !9fQiqAv@AA AAv 1yA~EFFG! ""$ TEQQZ'AE" q5QGJmmoFBA26)2.Aq&rN q&& g%% w<c#^^# TS:XaSv gTS-S;aSOSnTS:a TS:XaSOUmOb[TS - 5R5Vs/sH nTS - U-PM nn[UT5H m[UU4S jU55(dM O Tv [STS5H'n[ TS - U5S :XdM[ TUT5v M) gs snf7f) aGenerates the primitive roots for a prime ``p``. Explanation =========== The primitive roots generated are not necessarily sorted. However, the first one is the smallest primitive root. Find the element whose order is ``p-1`` from the smaller one. If we can find the first primitive root ``g``, we can use the following theorem. .. math :: \operatorname{ord}(g^k) = \frac{\operatorname{ord}(g)}{\gcd(\operatorname{ord}(g), k)} From the assumption that `\operatorname{ord}(g)=p-1`, it is a necessary and sufficient condition for `\operatorname{ord}(g^k)=p-1` that `\gcd(p-1, k)=1`. Parameters ========== p : odd prime Yields ====== int the primitive roots of ``p`` Examples ======== >>> from sympy.ntheory.residue_ntheory import _primitive_root_prime_iter >>> sorted(_primitive_root_prime_iter(19)) [2, 3, 10, 13, 14, 15] References ========== .. [1] W. Stein "Elementary Number Theory" (2011), page 44 N)r)rc3D># UHn[TUT5S:gv M g7frNr$).0pwgr)s r2 -_primitive_root_prime_iter..s2"3q"a=A%s )rkeysrangeallrr$)r)g_minivkrBs` @r2_primitive_root_prime_iterrL[sV AvQ&AaE2vbAe#,QU#3#8#8#: ;#:aa!e\#: ;uaA2222! G 1a^ q1ua=A aA,  >> from sympy.ntheory.residue_ntheory import _primitive_root_prime_power_iter >>> sorted(_primitive_root_prime_power_iter(5, 2)) [2, 3, 8, 12, 13, 17, 22, 23] rNr7rc34># UH nTT-U-v M g7fN)r@mrBrKs r2rC3_primitive_root_prime_power_iter..sF3EaA 3Es)rLr$rF)r)r*p2trBrKs @@r2 _primitive_root_prime_power_iterrUsD Av-a000 T+A.ASAE2&&",A1b_6F5AD"3EFFF%/ 1Gs(BB ABB<B =B Bc#`# [X5HnUS-S:XaUv MX U--v M g7f)aGenerates the primitive roots of `2p^e`. Explanation =========== If ``g`` is the primitive root of ``p**e``, then the odd one of ``g`` and ``g+p**e`` is the primitive root of ``2*p**e``. Parameters ========== p : odd prime e : positive integer Yields ====== int the primitive roots of `2p^e` Examples ======== >>> from sympy.ntheory.residue_ntheory import _primitive_root_prime_power2_iter >>> sorted(_primitive_root_prime_power2_iter(5, 2)) [3, 13, 17, 23, 27, 33, 37, 47] r7rN)rU)r)r*rBs r2!_primitive_root_prime_power2_iterrWs1:.a 3 q5A:Gd(N 4s,.c[U5nUS::a [S5eUS::aUS- $US-S:HnU(dUnOUS-(aUS-nOg[U5(aSnO)[US5nU(dgUup4[U5(dgU(d/U(a[ [ X455$[ [ X455$U(a3[SUS5H"nXc-(dM[X`5(dM Us $ [ [U55nUS:Xd[XsS- US-5S:waU$[US-U5H"nXc-(dM[X`5(dM Us $ g)a>Returns a primitive root of ``p`` or None. Explanation =========== For the definition of primitive root, see the explanation of ``is_primitive_root``. The primitive root of ``p`` exist only for `p = 2, 4, q^e, 2q^e` (``q`` is an odd prime). Now, if we know the primitive root of ``q``, we can calculate the primitive root of `q^e`, and if we know the primitive root of `q^e`, we can calculate the primitive root of `2q^e`. When there is no need to find the smallest primitive root, this property can be used to obtain a fast primitive root. On the other hand, when we want the smallest primitive root, we naively determine whether it is a primitive root or not. Parameters ========== p : integer, p > 1 smallest : if True the smallest primitive root is returned or None Returns ======= int | None : If the primitive root exists, return the primitive root of ``p``. If not, return None. Raises ====== ValueError If `p \le 1` or ``p`` is not an integer. Examples ======== >>> from sympy.ntheory.residue_ntheory import primitive_root >>> primitive_root(19) 2 >>> primitive_root(21) is None True >>> primitive_root(50, smallest=False) 27 See Also ======== is_primitive_root References ========== .. [1] W. Stein "Elementary Number Theory" (2011), page 44 .. [2] P. Hackman "Elementary Number Theory" (2009), Chapter C r%p should be an integer greater than 1r7rNr6) rr"rrnextrWrUrFis_primitive_rootrLr$)r)smallestp_evenqr*rQrIrBs r2primitive_rootr`s?| q AAv@AAAv1u UaZF   Q qDqzz  1a qzz  9!?@ @4Q:;; q!QAuu*100  ' *+AAvQAq!t$) 1q5!_ 55&q,,Hr4cZ^^^[T5[T5smmTS::a [S5eTT-m[TT5S:wa [S5eTS::aTTS- :H$TS-(aTnOTS-(aTS-nOg[U5(a"US- m[ US- 5R 5nOl[ US5nU(dgUup%[U5(dgX%S- -US- -m[[ US- 5R 55nURU5 [UUU4SjU55$) aReturns True if ``a`` is a primitive root of ``p``. Explanation =========== ``a`` is said to be the primitive root of ``p`` if `\gcd(a, p) = 1` and `\phi(p)` is the smallest positive number s.t. `a^{\phi(p)} \equiv 1 \pmod{p}`. where `\phi(p)` is Euler's totient function. The primitive root of ``p`` exist only for `p = 2, 4, q^e, 2q^e` (``q`` is an odd prime). Hence, if it is not such a ``p``, it returns False. To determine the primitive root, we need to know the prime factorization of ``q-1``. The hardness of the determination depends on this complexity. Parameters ========== a : integer p : integer, ``p`` > 1. ``a`` and ``p`` should be relatively prime Returns ======= bool : If True, ``a`` is the primitive root of ``p``. Raises ====== ValueError If `p \le 1` or `\gcd(a, p) \neq 1`. If ``a`` or ``p`` is not an integer. Examples ======== >>> from sympy.functions.combinatorial.numbers import totient >>> from sympy.ntheory import is_primitive_root, n_order >>> is_primitive_root(3, 10) True >>> is_primitive_root(9, 10) False >>> n_order(3, 10) == totient(10) True >>> n_order(9, 10) == totient(10) False See Also ======== primitive_root rrYr!rZr7Fr6c3J># UHn[TTU-T5S:gv M g7fr>r?)r@primer& group_orderr)s r2rC$is_primitive_root..s$Is1kU*A.!3s #) rr"rrrrErsetaddrG)r&r)r_r-rQr*rds`` @r2r\r\Ost !9fQiDAqAv@AA AA 1ayA~EFF AvAEz1u  Q qDqzz!e AE"'') 1a qzza%j!a%( iA&++-. A II IIr4c[US- 5nX- nUS-S:XaSnO.US-S;aSnO"[SUS- 5n[XA5S:XaOM![XU5n[XCU5nSn[ U5H@nU[XgU5-U-n [U S US- U- -U5n X-US- :XdM8US U-- nMB [XS-S -U5[XgS -U5-U-n U $) al Returns the square root in the case of ``p`` prime with ``p == 1 (mod 8)`` Assume that the root exists. Parameters ========== a : int p : int prime number. should be ``p % 8 == 1`` Returns ======= int : Generally, there are two roots, but only one is returned. Which one is returned is random. Examples ======== >>> from sympy.ntheory.residue_ntheory import _sqrt_mod_tonelli_shanks >>> _sqrt_mod_tonelli_shanks(2, 17) in [6, 11] True References ========== .. [1] Carl Pomerance, Richard Crandall, Prime Numbers: A Computational Perspective, 2nd Edition (2005), page 101, ISBN:978-0387252827 r r<r6)r7r6rr7)r rrr$rF) r&r)srTdADrQrIadmr1s r2_sqrt_mod_tonelli_shanksrqsB !a%A A2v{  Q& 1q5!Aa|r! A! A A! A A 1XA! nq #q1q519~q) 7a!e  AIA  AAz1c!T1o-1A Hr4cU(a[[X55$[[U55nUS-nSn[X5HnXS:aUs $XS:aX- s $UnM U$)aL Find a root of ``x**2 = a mod p``. Parameters ========== a : integer p : positive integer all_roots : if True the list of roots is returned or None Notes ===== If there is no root it is returned None; else the returned root is less or equal to ``p // 2``; in general is not the smallest one. It is returned ``p // 2`` only if it is the only root. Use ``all_roots`` only when it is expected that all the roots fit in memory; otherwise use ``sqrt_mod_iter``. Examples ======== >>> from sympy.ntheory import sqrt_mod >>> sqrt_mod(11, 43) 21 >>> sqrt_mod(17, 32, True) [7, 9, 23, 25] r7N)sorted sqrt_mod_iterabsr)r&r) all_rootshalfpr1rs r2sqrt_modrysc<mA)** F1IA FE A 1  9H Y5LA ! Hr4c # # [U5[[U55p/n/n[n[U5R 5HZupgX-(a [ XU5nO[ XU5n[nU(d gURU5 URXg-5 M\ [U5S:Xa[X#S5ShvN g[U[5upn U"U6Hn U"[XXU [55v M gN?7f)a~ Iterate over solutions to ``x**2 = a mod p``. Parameters ========== a : integer p : positive integer domain : integer domain, ``int``, ``ZZ`` or ``Integer`` Examples ======== >>> from sympy.ntheory.residue_ntheory import sqrt_mod_iter >>> list(sqrt_mod_iter(11, 43)) [21, 22] See Also ======== sqrt_mod : Same functionality, but you want a sorted list or only one solution. Nrr)rrurrr#_sqrt_mod_prime_power _sqrt_mod1rappendlenmapr r r) r&r)domainrJpv_productr/r0rxmmr*rlvxs r2rtrts0 !9c&)nq A BHA,$$& 6&qb1B A2&BH    "&' 1v{vt$$$2r?qA,B267 7 %sCDDADc,X-nX-nUS:XaUS-S:wagUS::a[[SUS55$Sn[SU5H#nUS-U- U- S-(dMUSUS- -- nM% SUS- -n[XCU- XF-U-XF-*U-/5$[X5S:wagUS-S:Xa[ XS-S-U5nOSUS-S:Xa?[ XS-S-U5n[ USU5X-:waU[ SUS- S-U5-U-nO [ X5nUS:atUn[UR 5S- 5H'n US-n[SU-U5n XwS-U- U -- U-nM) X"S- -(a[SU-U5n XwS-U- U -- U-n[XsU- /5$)a Find the solutions to ``x**2 = a mod p**k`` when ``a % p != 0``. If no solution exists, return ``None``. Solutions are returned in an ascending list. Parameters ========== a : integer p : prime number k : positive integer Examples ======== >>> from sympy.ntheory.residue_ntheory import _sqrt_mod_prime_power >>> _sqrt_mod_prime_power(11, 43, 1) [21, 22] References ========== .. [1] P. Hackman "Elementary Number Theory" (2009), page 160 .. [2] http://www.numbertheory.org/php/squareroot.html .. [3] [Gathen99]_ r7r8rNr6rZr<)listrFrsrr$rq bit_lengthr) r&r)rKpkrxnxhresr/_frinvs r2r{r{Fs6 B AAv q5A: 6aQ( (  1+BAb A%%Q26]" !a%Lqq&15B,!%2 >??a|q1uz!!e\1% Q!!!e\1% sAq>QU "AA!|Q//!3C&q,1uq||~)*AQB1S5"%Eq&1*e++r1C+ A;1S5"%Eq&1*e++r1C 3S/ ""r4c^^^^TT-mUT-nUS:Xa[STTTS-S--5$[UT5upUS-S:Xag[UTTU- 5nUcgUS-mUUUU4SjU5$)a Find solution to ``x**2 == a mod p**n`` when ``a % p == 0``. If no solution exists, return ``None``. Parameters ========== a : integer p : prime number, p must divide a n : positive integer References ========== .. [1] http://www.numbertheory.org/php/squareroot.html rrr7Nc3h># UH'n[UTT--TTTT- -5Ho"v M M) g7frOrF)r@rr1rQr'r)pns r2rC_sqrt_mod1..s3 D"U2ad7BAE %CA%CAs/2)rFr r{)r&r)r'rxrrQrs `` @@r2r|r|s" AB BAAvQAQ1 -.. !Q>> from sympy.ntheory import is_quad_residue >>> is_quad_residue(21, 100) True Indeed, ``pow(39, 2, 100)`` would be 21. >>> is_quad_residue(21, 120) False That is, for any integer ``x``, ``pow(x, 2, 120)`` is not 21. If ``p`` is an odd prime, an iterative method is used to make the determination: >>> from sympy.ntheory import is_quad_residue >>> sorted(set([i**2 % 7 for i in range(7)])) [0, 1, 2, 4] >>> [j for j in range(7) if is_quad_residue(j, 7)] [0, 1, 2, 4] See Also ======== legendre_symbol, jacobi_symbol, sqrt_mod rz p must be > 0r7r6TrjFrkr)rr"r rrrr#r )r&r)rTa_rxjr/r0s r2is_quad_residuers%f !9fQiq1u))FA1uA ! A!q&\ " A1uA    q5AEq ABw'!**Av A,$$& 6a}!"VBQw2NEB1ur!+' r4cV^^TU-m[T5[T5[U5smmnUS::a [S5eTS:a [S5eTS:Xa US:XagTS:H$TS:XagTS:XagTS:Xa [TU5$[UU4Sj[ U5R 555$) z Returns True if ``x**n == a (mod m)`` has solutions. References ========== .. [1] P. Hackman "Elementary Number Theory" (2009), page 76 rz m must be > 0zn must be >= 0rFTr7c3B># UHup[TTX5v M g7frO)#_is_nthpow_residue_bign_prime_power)r@r)r*r&r's r2rC$is_nthpow_residue..'s&1/4131a>>/s)rr"rrGrr#)r&r'rQs`` r2is_nthpow_residuer s AAQiF1IGAq!Av))1u)**Av 6Av AvAvAvq!$$ 1$Q<--/1 11r4c`X-S:Xa;U[X#5-nU(dg[X5upXA-(agX4-nX-S:XaM;US:wa1X#S- -US- -n[X[XQ5-[X#55S:H$US-(ag[[ U5S-U5nU[SU5-S:H$)z Returns True if `x^n = a \pmod{p^k}` has solutions for `n > 2`. Parameters ========== a : positive integer n : integer, n > 2 p : prime number k : positive integer rTFr7r)r$r rminr )r&r'r)rKmufcs r2rr+s %1* SYq  6  %1* Av AJA 13q9nc!i0A551u IaL1 a A s1ay=A r4c r[[U55nUn[U5R5HupgUS- Xg--nX-S:XaX-nX-S:XaMU[ U*U5-n SU -U-n [ U[ XXU5[ XHU-U55n [U5H-n [ XZU5[ XI*U -US- -U5-U-nX-n M/ M U/n [ XBS- U-U5nUn[US- 5Hn X-U-nU RU5 M U(aU R5 U $[U 5$)a Root of ``x**q = s mod p``, ``p`` prime and ``q`` divides ``p - 1``. Assume that the root exists. Parameters ========== s : integer q : integer, n > 2. ``q`` divides ``p - 1``. p : prime number all_roots : if False returns the smallest root, else the list of roots Returns ======= list[int] | int : Root of ``x**q = s mod p``. If ``all_roots == True``, returned ascending list. otherwise, returned an int. Examples ======== >>> from sympy.ntheory.residue_ntheory import _nthroot_mod1 >>> _nthroot_mod1(5, 3, 13, False) 7 >>> _nthroot_mod1(13, 4, 17, True) [3, 5, 12, 14] References ========== .. [1] A. M. Johnston, A Generalized qth Root Algorithm, ACM-SIAM Symposium on Discrete Algorithms (1999), pp. 929-930 rr) r[rLrr#r discrete_logr$rFr}sortr)rlr_r)rvrBrxqxr0rzr1rTrrrhxs r2 _nthroot_mod1rIs@H ' *+A AA,$$& Urv fk HAfk faRn  UrM CaL#a2q/ :rAA! SBqDAENA66:A HA' #C AA!|QA B 1q5\daZ 2   s8Or4cv[XX#5(d/$X-nUS:XaS/nOUS- U-S:Xa [XAUSS9nOoUnUS- nSnXg:aXXF4upFpU(a-[Xg5up[X*U5U-U-n XzpvXpU(aM-US:XaU/nOUS:Xa [ XBSS9nO [XFUSS9nUS:XaU$XU--n[ 5n UHn [XS- U5U-U-nUnUS:waS[ X5n[US- 5H#nX-n[XU5U- nUU-nU U- U-n M% U RU 5 MsU 1n[US- 5H[nX-n[ 5nUHBn[UX5X-:waMUU;dMURU5 UX--U-nUU;aM#MD UnM] U U-n M [U 5$)zRoot of ``x**n = a mod p**k``. Parameters ========== a : integer n : integer, n > 2 p : prime number k : positive integer Returns ======= list[int] : Ascending list of roots of ``x**n = a mod p**k``. If no solution exists, return ``[]``. rrT)rvr7) rrdivmodr$ryrfrrFrgrs)r&r'r)rKa_mod_p base_rootspapbbr_pcr tot_rootsrootdiffnew_basem_invrtmp roots_in_base new_rootsk_s r2_nthroot_mod_prime_powerrs& /qQ : : eG!|S a%1 "7qDA  U  7!"!3 G2NEAAr1 '!+AQ b 7!J 1W!'=J&wAFJAvAIAI4Q"1$q( 194OE1q5\ $8,q0u s h. " MM$ !FM1q5\ E 'B2q+q|; I-! b) HM2h>I-( !* ""M1I/0 ) r4cX-n[U5[U5[U5p!nUS:a [S5eUS:a [S5eUS:Xa U(aU/$U$US:Xa [XU5$/n/n[U5R 5HTupg[ XXg5nU(dU(a/s $Ss $UR Xg-5 UR [U55 MV [U[5upn [[[[U6V s1sHn [XXU [5iM sn 55n U(aU $U (aU S$gs sn f)a Find the solutions to ``x**n = a mod p``. Parameters ========== a : integer n : positive integer p : positive integer all_roots : if False returns the smallest root, else the list of roots Returns ======= list[int] | int | None : solutions to ``x**n = a mod p``. The table of the output type is: ========== ========== ========== all_roots has roots Returns ========== ========== ========== True Yes list[int] True No [] False Yes int False No None ========== ========== ========== Raises ====== ValueError If ``a``, ``n`` or ``p`` is not integer. If ``n`` or ``p`` is not positive. Examples ======== >>> from sympy.ntheory.residue_ntheory import nthroot_mod >>> nthroot_mod(11, 4, 19) 8 >>> nthroot_mod(11, 4, 19, True) [8, 11] >>> nthroot_mod(68, 3, 109) 23 References ========== .. [1] P. Hackman "Elementary Number Theory" (2009), page 76 rn should be positivezp should be positiver7Nr)rr"ryrr#rr}rsr r rr%rr)r&r'r)rvbase prime_powerr_r*rPESrrets r2 nthroot_modrs@h AQiF1I!A1u/001u/00Avs&Q&Avi(( DK! ""$,Q18 "2 , ,14  F9%& % k2&GA! S$+TN4$2q#11B?$245 6C  1v  4s E c[U5n[US-S-5Vs1sHn[USU5iM nn[U5$s snf)z Returns the list of quadratic residues. Examples ======== >>> from sympy.ntheory.residue_ntheory import quadratic_residues >>> quadratic_residues(7) [0, 1, 2, 4] r7r)rrFr$rs)r)rIrxs r2quadratic_residuesr%sG q A$Q!VaZ010!Q10A1 !9 2sAz~The `sympy.ntheory.residue_ntheory.legendre_symbol` has been moved to `sympy.functions.combinatorial.numbers.legendre_symbol`.z1.13z%deprecated-ntheory-symbolic-functions)deprecated_since_versionactive_deprecations_targetcSSKJn U"X5$)a Returns the Legendre symbol `(a / p)`. .. deprecated:: 1.13 The ``legendre_symbol`` function is deprecated. Use :class:`sympy.functions.combinatorial.numbers.legendre_symbol` instead. See its documentation for more information. See :ref:`deprecated-ntheory-symbolic-functions` for details. For an integer ``a`` and an odd prime ``p``, the Legendre symbol is defined as .. math :: \genfrac(){}{}{a}{p} = \begin{cases} 0 & \text{if } p \text{ divides } a\\ 1 & \text{if } a \text{ is a quadratic residue modulo } p\\ -1 & \text{if } a \text{ is a quadratic nonresidue modulo } p \end{cases} Parameters ========== a : integer p : odd prime Examples ======== >>> from sympy.functions.combinatorial.numbers import legendre_symbol >>> [legendre_symbol(i, 7) for i in range(7)] [0, 1, 1, -1, 1, -1, -1] >>> sorted(set([i**2 % 7 for i in range(7)])) [0, 1, 2, 4] See Also ======== is_quad_residue, jacobi_symbol r)legendre_symbol)%sympy.functions.combinatorial.numbersr)r&r)_legendre_symbols r2rr5sZZ A !!r4zzThe `sympy.ntheory.residue_ntheory.jacobi_symbol` has been moved to `sympy.functions.combinatorial.numbers.jacobi_symbol`.cSSKJn U"X5$)a! Returns the Jacobi symbol `(m / n)`. .. deprecated:: 1.13 The ``jacobi_symbol`` function is deprecated. Use :class:`sympy.functions.combinatorial.numbers.jacobi_symbol` instead. See its documentation for more information. See :ref:`deprecated-ntheory-symbolic-functions` for details. For any integer ``m`` and any positive odd integer ``n`` the Jacobi symbol is defined as the product of the Legendre symbols corresponding to the prime factors of ``n``: .. math :: \genfrac(){}{}{m}{n} = \genfrac(){}{}{m}{p^{1}}^{\alpha_1} \genfrac(){}{}{m}{p^{2}}^{\alpha_2} ... \genfrac(){}{}{m}{p^{k}}^{\alpha_k} \text{ where } n = p_1^{\alpha_1} p_2^{\alpha_2} ... p_k^{\alpha_k} Like the Legendre symbol, if the Jacobi symbol `\genfrac(){}{}{m}{n} = -1` then ``m`` is a quadratic nonresidue modulo ``n``. But, unlike the Legendre symbol, if the Jacobi symbol `\genfrac(){}{}{m}{n} = 1` then ``m`` may or may not be a quadratic residue modulo ``n``. Parameters ========== m : integer n : odd positive integer Examples ======== >>> from sympy.functions.combinatorial.numbers import jacobi_symbol, legendre_symbol >>> from sympy import S >>> jacobi_symbol(45, 77) -1 >>> jacobi_symbol(60, 121) 1 The relationship between the ``jacobi_symbol`` and ``legendre_symbol`` can be demonstrated as follows: >>> L = legendre_symbol >>> S(45).factors() {3: 2, 5: 1} >>> jacobi_symbol(7, 45) == L(7, 3)**2 * L(7, 5)**1 True See Also ======== is_quad_residue, legendre_symbol r) jacobi_symbol)rr)rQr'_jacobi_symbols r2rrfsFV ! r4zlThe `sympy.ntheory.residue_ntheory.mobius` has been moved to `sympy.functions.combinatorial.numbers.mobius`.cSSKJn U"U5$)a> Mobius function maps natural number to {-1, 0, 1} .. deprecated:: 1.13 The ``mobius`` function is deprecated. Use :class:`sympy.functions.combinatorial.numbers.mobius` instead. See its documentation for more information. See :ref:`deprecated-ntheory-symbolic-functions` for details. It is defined as follows: 1) `1` if `n = 1`. 2) `0` if `n` has a squared prime factor. 3) `(-1)^k` if `n` is a square-free positive integer with `k` number of prime factors. It is an important multiplicative function in number theory and combinatorics. It has applications in mathematical series, algebraic number theory and also physics (Fermion operator has very concrete realization with Mobius Function model). Parameters ========== n : positive integer Examples ======== >>> from sympy.functions.combinatorial.numbers import mobius >>> mobius(13*7) 1 >>> mobius(1) 1 >>> mobius(13*7*5) -1 >>> mobius(13**2) 0 References ========== .. [1] https://en.wikipedia.org/wiki/M%C3%B6bius_function .. [2] Thomas Koshy "Elementary Number Theory with Applications" r)mobius)rr)r'_mobiuss r2rrsdH 1:r4NczX-nX -nUcUnSn[U5HnXA:XaUs $XB-U-nM [S5e)ag Trial multiplication algorithm for computing the discrete logarithm of ``a`` to the base ``b`` modulo ``n``. The algorithm finds the discrete logarithm using exhaustive search. This naive method is used as fallback algorithm of ``discrete_log`` when the group order is very small. The value ``n`` must be greater than 1. Examples ======== >>> from sympy.ntheory.residue_ntheory import _discrete_log_trial_mul >>> _discrete_log_trial_mul(41, 15, 7) 3 See Also ======== discrete_log References ========== .. [1] "Handbook of applied cryptography", Menezes, A. J., Van, O. P. C., & Vanstone, S. A. (1997). rLog does not exist)rFr")r'r&rr.r1rIs r2_discrete_log_trial_mulrsR6FAFA } A 5\ 6H EAI ) **r4cX-nX -nUc [X 5n[U5S-n0nSn[U5HnXuU'Xb-U-nM [X$*U5nUn[U5HnXe;a Xt-XV-s $Xh-U-nM [ S5e)a! Baby-step giant-step algorithm for computing the discrete logarithm of ``a`` to the base ``b`` modulo ``n``. The algorithm is a time-memory trade-off of the method of exhaustive search. It uses `O(sqrt(m))` memory, where `m` is the group order. Examples ======== >>> from sympy.ntheory.residue_ntheory import _discrete_log_shanks_steps >>> _discrete_log_shanks_steps(41, 15, 7) 3 See Also ======== discrete_log References ========== .. [1] "Handbook of applied cryptography", Menezes, A. J., Van, O. P. C., & Vanstone, S. A. (1997). rr)r3rrFr$r") r'r&rr.rQTr1rIrs r2_discrete_log_shanks_stepsr s4FAFA }  U aA A A 1X! EAI Ar1 A A 1X 6514<  EAI ) **r4cX-nX -nUc [X 5n[U5n[U5GHnU"SUS- 5nU"SUS- 5n [X(U5[XU5-U-n U S-n U S:XaX-U-n Un U S-U-nO-U S:XaX-U-n X-U-n X-U-nOX*-U-n US-U-n U n[U5GHnU S-n U S:XaX-U-n U S-U-n O+U S:XaX-U-n X-U-nX-U-n OX*-U-n US-U-nU S-n U S:XaX-U-n US-U-nO+U S:XaX-U-n X-U-n X-U-nOX,-U-n U S-U-n U S-n U S:XaX-U-n US-U-nO+U S:XaX-U-n X-U-n X-U-nOX,-U-n U S-U-n X:XdMX- U-n[ UU5X- -U-n[UUU5U- U-S:XaUs s $ GM GM [ S5e![ a N"f=f)a# Pollard's Rho algorithm for computing the discrete logarithm of ``a`` to the base ``b`` modulo ``n``. It is a randomized algorithm with the same expected running time as ``_discrete_log_shanks_steps``, but requires a negligible amount of memory. Examples ======== >>> from sympy.ntheory.residue_ntheory import _discrete_log_pollard_rho >>> _discrete_log_pollard_rho(227, 3**7, 3) 7 See Also ======== discrete_log References ========== .. [1] "Handbook of applied cryptography", Menezes, A. J., Van, O. P. C., & Vanstone, S. A. (1997). rr6rz&Pollard's Rho failed to find logarithm)r3rrFr$rZeroDivisionErrorr")r'r&rr.retriesrseedrrIaabaxarxbabbbrrxr*s r2_discrete_log_pollard_rhor7s4FAFA } uoG 7^ Q " Q " ]S] *Q . F 6!BBq&E!B !V1B'U"B'U"B!Bq&E!BBuAQAAvVaZ1f%aWq[g&g&VaZ1f%QAAvVaZ1f%aWq[g&g&VaZ1f%QAAvVaZ1f%aWq[g&g&VaZ1f%xW%q%(BG4u>)s,G'' G43G4cS/[U5-n[U5H)up4X-S:XdMX#==S- ss'X-nX-S:XaMM+ US:wagU$)zTry to factor n with respect to a given factorbase. Upon success a list of exponents with respect to the factorbase is returned. Otherwise None.rrN)r~ enumerate)r' factorbaser-rIr)s r2_discrete_log_is_smoothrs\c#j/!G*%eqj J!OJAeqj& Av Nr4c >[U5nSSKJnJnJn X-nX -n[ U"SU"U"U5U"U"U55-5-SSU"U"U55- --55n SU -U -n [ [U 55n [U 5n US- n Un[U5HGnUS:Xa X?- U-s $[X5nU(aUVs/sHnUU-PM snU/-n OX-U-nMI [S5eS/U -nSnSnUSU -:GaUU :GaU"SU 5n[[X/U5U 5nUcUS- nM:US- nSnUU/- nU n[U 5H^nUUU-nUS:a3UUb-[U S-5HnUUUUUU-- U-UU'M OUUU'UUS:dMTUU :XdM\UnM` UU :XdUUbM[UUS U5n[UU S-5HnUUU-U-UU'M UUU'[U 5HNnUUS:a7UUb1UUn[U S-5HnUUUUUU-- U-UU'M UUS:dMN O* UUU - U-n[X/U5U:XaU$[S5eUSU -:a UU :aGM[S5es snf) aP Index Calculus algorithm for computing the discrete logarithm of ``a`` to the base ``b`` modulo ``n``. The group order must be given and prime. It is not suitable for small orders and the algorithm might fail to find a solution in such situations. Examples ======== >>> from sympy.ntheory.residue_ntheory import _discrete_log_index_calculus >>> _discrete_log_index_calculus(24570203447, 23859756228, 2, 12285101723) 4519867240 See Also ======== discrete_log References ========== .. [1] "Handbook of applied cryptography", Menezes, A. J., Van, O. P. C., & Vanstone, S. A. (1997). r)rexplogg?rr<zIndex Calculus failedNr6rk) rmathrrrr%rrr~rFrr"r$)r'r&rr.rrrrrBmaxrlfordermoabxr1 relationarx relationsrKkkrelationindexrIrirrinvrbis r2_discrete_log_index_calculusrs"4uoG##FAFA CdCFSQ[022Q3s1v;5FH IJA a%!)Cjm$J ZBAgG C 5\ !8I& &+C< ,56IqUI6!>> from sympy.ntheory.residue_ntheory import _discrete_log_pohlig_hellman >>> _discrete_log_pohlig_hellman(251, 210, 71) 197 See Also ======== discrete_log References ========== .. [1] "Handbook of applied cryptography", Menezes, A. J., Van, O. P. C., & Vanstone, S. A. (1997). rrrT) modularrr3rr~rr#rFr$r)r'r&rr. order_factorsrlrIpirrajbjcjrmrs r2_discrete_log_pohlig_hellmanr s6FAFA } !%( c-  A !4!4!67 8BrAQQ1q))5QK+?CBQ Q'BaRT2B DBQJ D 8 ]%8%8%:;%:62%:;Q ?DA HX- X'MF Uc [U5nUS:a [XX#5$U(aWSU"U"U5U"U"U55-5-U"U5S- :a [XX#5$US:a [XX#5$[XX#5$[XX#W 5$) a Compute the discrete logarithm of ``a`` to the base ``b`` modulo ``n``. This is a recursive function to reduce the discrete logarithm problem in cyclic groups of composite order to the problem in cyclic groups of prime order. It employs different algorithms depending on the problem (subgroup order size, prime order or not): * Trial multiplication * Baby-step giant-step * Pollard's Rho * Index Calculus * Pohlig-Hellman Examples ======== >>> from sympy.ntheory import discrete_log >>> discrete_log(41, 15, 7) 3 References ========== .. [1] https://mathworld.wolfram.com/DiscreteLogarithm.html .. [2] "Handbook of applied cryptography", Menezes, A. J., Van, O. P. C., & Vanstone, S. A. (1997). r)rrrrirZ lJ))rrrrr"rr#rFr$rrrrrr)r'r&rr. prime_orderrrr-r/kxpykyrr)r*rIrs r2rr;s@QiF1I!A1u/00Av }l((*FBAv=K26)K"$q&GK#BF+113=K2%K"$BK 4 +mmoFB VOE& MMODAA1Xq1*a(A-KEFA  u#$5  $en  t|&qQ66  T#a&SV$ % %E R 7/a? ? ] "-aA= =(q88 'a FFr4c^^[U5n[T5m[U5n[T5mTS::a [S5eUT-nTT-mUT-nUS:Xa[TU*T5$TS:Xa9/nUS:XaURS5 TU-S-(aURS5 U$[ SU-T5S:XaO[ UT5nTU-mX%-nTS-(aTT- mTS-m[ UU4Sj[TS-U- T555$[5n[TS-SU-U-- SU-T-5H3n[UT- SU-5upU S:XdMURUT-5 M5 [ U5$)a  Find the solutions to `a x^2 + b x + c \equiv 0 \pmod{n}`. Parameters ========== a : int b : int c : int n : int A positive integer. Returns ======= list[int] : A sorted list of solutions. If no solution exists, ``[]``. Examples ======== >>> from sympy.ntheory.residue_ntheory import quadratic_congruence >>> quadratic_congruence(2, 5, 3, 7) # 2x^2 + 5x + 3 = 0 (mod 7) [2, 6] >>> quadratic_congruence(8, 6, 4, 15) # No solution [] See Also ======== polynomial_congruence : Solve the polynomial congruence rr rr7c32># UH oT- T-v M g7frOrP)r@rIrr's r2rC'quadratic_congruence..sF+Ea1uk+EsrZ) rr"rr}rrrsrtrfrrg) r&rrr'rootsinv_arrIr_rems ` ` r2quadratic_congruencerswD q Aq Aq Aq AAv@AAFAFAFAAv QB**Av 6 LLO EQ; LLO  1Q3{aq!  U   q5 FA aF=A1+EFFF %C 1a4!A#a%<1Q /Aqs# !8 GGAEN0 #;r4cUR5(d [S5e[U5nUR(d [S5eUR[ :Xd [S5eUR 5$)zq return coefficients of expr if it is a univariate polynomial with integer coefficients else raise a ValueError. z%The expression should be a polynomialz#The expression should be univariatez6The expression should should have integer coefficients) is_polynomialr"r is_univariaterr all_coeffs)expr polynomials r2 _valid_exprrsf     @AAdJ  # #>??    "QRR  ""r4c*[U5nUVs/sHo3U-PM nn[U5nUS:Xa [/UQUP76$US:Xa[S/UQUP76$USS:Xa*SUS-[U5:Xa[ US*US- US5$[ X!5$s snf)a Find the solutions to a polynomial congruence equation modulo m. Parameters ========== expr : integer coefficient polynomial m : positive integer Examples ======== >>> from sympy.ntheory import polynomial_congruence >>> from sympy.abc import x >>> expr = x**6 - 2*x**5 -35 >>> polynomial_congruence(expr, 6125) [3257] See Also ======== sympy.polys.galoistools.gf_csolve : low level solving routine used by this routine r6r7rrrkT)rr~rsumrr)rrQ coefficientsnumranks r2polynomial_congruencers2t$L'34|!G|L4 | D qy#5\5155 qy#A8 8a88A!L$4 4L8I IL,,dQh4@@ \ %%5sBc 8US:a [S5eUS:d US:dX:ag[U5nUR5VVs/sHupE[XXE5PM nnn[ UR5VVs/sH upGXG-PM snnUSS9S$s snnfs snnf)aCompute ``binomial(n, m) % k``. Explanation =========== Returns ``binomial(n, m) % k`` using a generalization of Lucas' Theorem for prime powers given by Granville [1]_, in conjunction with the Chinese Remainder Theorem. The residue for each prime power is calculated in time O(log^2(n) + q^4*log(n)log(p) + q^4*p*log^3(p)). Parameters ========== n : an integer m : an integer k : a positive integer Examples ======== >>> from sympy.ntheory.residue_ntheory import binomial_mod >>> binomial_mod(10, 2, 6) # binomial(10, 2) = 45 3 >>> binomial_mod(17, 9, 10) # binomial(17, 9) = 24310 0 References ========== .. [1] Binomial coefficients modulo prime powers, Andrew Granville, Available: https://web.archive.org/web/20170202003812/http://www.dms.umontreal.ca/~andrew/PDF/BinCoeff.pdf rzk is required to be positiverF)check)r"rr#_binomial_mod_prime_powerr)r'rQrK factorisationr)r*residuesrAs r2 binomial_modr!sB 1uJ=>>  1uAqaLMBOBUBUBWXBW$!)!5BWHX =#6#6#89#8%!#8985 QRS TTY9s B5B c,^^^^^^^[TT5mUUU4SjmSmUUUU4Sjm[S/5U4Sj5mUUUUU4SjnSnXX- pnS=n =n =pU(aU"UT-5n U"UT-5U"UT-5-T-nUT-[UT5[UT5snupunnUU-U -T-n X- n U TS- :aX- n X][UT5--nUT-nU S- n U(aM[TS:XaTS :aSOS U T5n[TU T5U-U-T-$) ahCompute ``binomial(n, m) % p**q`` for a prime ``p``. Parameters ========== n : positive integer m : a nonnegative integer p : a prime q : a positive integer (the prime exponent) Examples ======== >>> from sympy.ntheory.residue_ntheory import _binomial_mod_prime_power >>> _binomial_mod_prime_power(10, 2, 3, 2) # binomial(10, 2) = 45 0 >>> _binomial_mod_prime_power(17, 9, 2, 4) # binomial(17, 9) = 24310 6 References ========== .. [1] Binomial coefficients modulo prime powers, Andrew Granville, Available: https://web.archive.org/web/20170202003812/http://www.dms.umontreal.ca/~andrew/PDF/BinCoeff.pdf c@>T S-nS=p#US:XaT S:XdSU-S-T T T -4;a/T S-S:XaUS- n[T SU-5[T SU-T - 5pTO&[T SU-S-5[T SU-S-T - 5pT[SUS-5HFn[US- T -S-UT -5H nX'-nX$-nM T "XU5nU[X(U5-nX4-nMH T S:Xa:US-n [SUS-5HoiUS-T "XU5-- n M U S-S:XaUS-nX4U--nX4-$)zCompute (u*p)!_p modulo p^q.r7rrk)r$rF) urxfacprodmodulodivrmulbj_smrr)r_s r2 up_factorial/_binomial_mod_prime_power..up_factorial^sY F 6a1f!aAqs8 31uz161a1+s1acAgCa1q/3q1Q37a-+@Cq!a%Aa!eQY]AaC0  1Q1+C C&) )D ND ! 6aB1a!e_AqD2aA;,>&>b_Av{DBJD  }r4cUn[SUS-5HnXA:wdM X0U-XD-- -nM [SUS-5HnXA:wdM X1U-XD-- -nM X1-$)z@Compute the exponent of (j*p)!_p in the calculation of (u*p)!_p.rr)r$rrxr&rIs r2r%_binomial_mod_prime_power..bjuscq!a%AvtsQSy(t!q!a%Avt!ac )t!yr4ch>Sn[T "U5T 5n[ST 5HnUn[UT -S-UT -U-S-5H nXV-nUT -nM Un[ST 5HnX:wdM XpU- -nM [ST 5HnX:wdM XtU- -nM Xt-nU[XWT 5-nUT -nM U$)z*Compute binomial(u*p + v, v)_p modulo p^q.r)rrFr$) r$rJr&r(rrv_rrI factorialr'r)r_s r2up_plus_v_binom2_binomial_mod_prime_power..up_plus_v_binom~sYq\6*q!AAAaC!GQqS1Wq[1V 2B1a[62Q;2!1a[62a%<2! HB Cv& &D FND r4rc>XS-T-$)zCompute v! modulo p^q.rkrP)rJprevr's r2r2,_binomial_mod_prime_power..factorialsbzF""r4cZ>[UT5upT"U5T"U5-T"X5-T-$)zCompute n!_p modulo p^q.)r)r'r$rJr2r'r)r,r3s r2 factorial_p._binomial_mod_prime_power..factorial_ps2a|! |A.1FF&PPr4rr7r6rk)r$rrr)r'rQr)r_r9r&NjMjRje0carryeq_1r numerator denominatormjrjr)rr2r'r,r3s `` @@@@@r2rr>sZ<AYF.&aS##QQ DquBBB V , !"v+.R&[1IIFR !#Qr1 vb!}HRhr2b5Q&  A:t}t F;777  Q " 16a1fa"dF ;C 2v  $t +v 55r4)T)F)returnz list[int]rO)NrN)r'r%rr)NN)K __future__rsympy.external.gmpyrrrrrr r sympy.polysr sympy.polys.domainsr sympy.polys.galoistoolsr rrr primetestrgeneraterfactor_rrrrsympy.utilities.decoratorrsympy.utilities.memoizationrsympy.utilities.miscrsympy.utilities.iterablesrsympy.core.randomrr itertoolsrr3rLrUrWr`r\rqryr%rtr{r|rrrrrrrrrrrrrrrrrrrrr!rrPr4r2rTs"444"RR .07'./CL=@*GZ!H`FXJv: z* Z #/8dH#VEBXv1<<;|K\M`    BBD*" D*"Z  ~BD@ D@ F  pBD/ D/d$+N*+Zb?J ^.B- `WGvDN # "&J)UXm6r4