jacobi.c
上传用户:qaz666999
上传日期:2022-08-06
资源大小:2570k
文件大小:10k
源码类别:

数学计算

开发平台:

Unix_Linux

  1. /* mpz_jacobi, mpz_legendre, mpz_kronecker -- mpz/mpz Jacobi symbols.
  2. Copyright 2000, 2001, 2002, 2005 Free Software Foundation, Inc.
  3. This file is part of the GNU MP Library.
  4. The GNU MP Library is free software; you can redistribute it and/or modify it
  5. under the terms of the GNU Lesser General Public License as published by the
  6. Free Software Foundation; either version 3 of the License, or (at your
  7. option) any later version.
  8. The GNU MP Library is distributed in the hope that it will be useful, but
  9. WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  10. FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License
  11. for more details.
  12. You should have received a copy of the GNU Lesser General Public License along
  13. with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  14. #include <stdio.h>
  15. #include "gmp.h"
  16. #include "gmp-impl.h"
  17. #include "longlong.h"
  18. /* Change this to "#define TRACE(x) x" for some traces. */
  19. #define TRACE(x)
  20. #define MPN_RSHIFT_OR_COPY(dst,src,size,shift)                  
  21.   do {                                                          
  22.     if ((shift) != 0)                                           
  23.       {                                                         
  24.         ASSERT_NOCARRY (mpn_rshift (dst, src, size, shift));    
  25.         (size) -= ((dst)[(size)-1] == 0);                       
  26.       }                                                         
  27.     else                                                        
  28.       MPN_COPY (dst, src, size);                                
  29.   } while (0)
  30. /* This code does triple duty as mpz_jacobi, mpz_legendre and mpz_kronecker.
  31.    mpz_jacobi could assume b is odd, but the improvements from that seem
  32.    small compared to other operations, and anything significant should be
  33.    checked at run-time since we'd like odd b to go fast in mpz_kronecker
  34.    too.
  35.    mpz_legendre could assume b is an odd prime, but knowing this doesn't
  36.    present any obvious benefits.  Result 0 wouldn't arise (unless "a" is a
  37.    multiple of b), but the checking for that takes little time compared to
  38.    other operations.
  39.    The main loop is just a simple binary GCD with the jacobi symbol result
  40.    tracked during the reduction.
  41.    The special cases for a or b fitting in one limb let mod_1 or modexact_1
  42.    get used, without any copying, and end up just as efficient as the mixed
  43.    precision mpz_kronecker_ui etc.
  44.    When tdiv_qr is called it's not necessary to make "a" odd or make a
  45.    working copy of it, but tdiv_qr is going to be pretty slow so it's not
  46.    worth bothering trying to save anything for that case.
  47.    Enhancements:
  48.    mpn_bdiv_qr should be used instead of mpn_tdiv_qr.
  49.    Some sort of multi-step algorithm should be used.  The current subtract
  50.    and shift for every bit is very inefficient.  Lehmer (per current gcdext)
  51.    would need some low bits included in its calculation to apply the sign
  52.    change for reciprocity.  Binary Lehmer keeps low bits to strip twos
  53.    anyway, so might be better suited.  Maybe the accelerated GCD style k-ary
  54.    reduction would work, if sign changes due to the extra factors it
  55.    introduces can be accounted for (or maybe they can be ignored).  */
  56. int
  57. mpz_jacobi (mpz_srcptr a, mpz_srcptr b)
  58. {
  59.   mp_srcptr  asrcp, bsrcp;
  60.   mp_size_t  asize, bsize;
  61.   mp_ptr     ap, bp;
  62.   mp_limb_t  alow, blow, ahigh, bhigh, asecond, bsecond;
  63.   unsigned   atwos, btwos;
  64.   int        result_bit1;
  65.   TMP_DECL;
  66.   TRACE (printf ("start asize=%d bsize=%dn", SIZ(a), SIZ(b));
  67.          mpz_trace (" a", a);
  68.          mpz_trace (" b", b));
  69.   asize = SIZ(a);
  70.   asrcp = PTR(a);
  71.   alow = asrcp[0];
  72.   bsize = SIZ(b);
  73.   if (bsize == 0)
  74.     return JACOBI_LS0 (alow, asize);  /* (a/0) */
  75.   bsrcp = PTR(b);
  76.   blow = bsrcp[0];
  77.   if (asize == 0)
  78.     return JACOBI_0LS (blow, bsize);  /* (0/b) */
  79.   /* (even/even)=0 */
  80.   if (((alow | blow) & 1) == 0)
  81.     return 0;
  82.   /* account for effect of sign of b, then ignore it */
  83.   result_bit1 = JACOBI_BSGN_SS_BIT1 (asize, bsize);
  84.   bsize = ABS (bsize);
  85.   /* low zero limbs on b can be discarded */
  86.   JACOBI_STRIP_LOW_ZEROS (result_bit1, alow, bsrcp, bsize, blow);
  87.   count_trailing_zeros (btwos, blow);
  88.   TRACE (printf ("b twos %un", btwos));
  89.   /* establish shifted blow */
  90.   blow >>= btwos;
  91.   if (bsize > 1)
  92.     {
  93.       bsecond = bsrcp[1];
  94.       if (btwos != 0)
  95.         blow |= (bsecond << (GMP_NUMB_BITS - btwos)) & GMP_NUMB_MASK;
  96.     }
  97.   /* account for effect of sign of a, then ignore it */
  98.   result_bit1 ^= JACOBI_ASGN_SU_BIT1 (asize, blow);
  99.   asize = ABS (asize);
  100.   if (bsize == 1 || (bsize == 2 && (bsecond >> btwos) == 0))
  101.     {
  102.       /* special case one limb b, use modexact and no copying */
  103.       /* (a/2)=(2/a) with a odd, and if b is even then a is odd here */
  104.       result_bit1 ^= JACOBI_TWOS_U_BIT1 (btwos, alow);
  105.       if (blow == 1)   /* (a/1)=1 always */
  106.         return JACOBI_BIT1_TO_PN (result_bit1);
  107.       JACOBI_MOD_OR_MODEXACT_1_ODD (result_bit1, alow, asrcp, asize, blow);
  108.       TRACE (printf ("base (%lu/%lu) with %dn",
  109.                      alow, blow, JACOBI_BIT1_TO_PN (result_bit1)));
  110.       return mpn_jacobi_base (alow, blow, result_bit1);
  111.     }
  112.   /* Discard low zero limbs of a.  Usually there won't be anything to
  113.      strip, hence not bothering with it for the bsize==1 case.  */
  114.   JACOBI_STRIP_LOW_ZEROS (result_bit1, blow, asrcp, asize, alow);
  115.   count_trailing_zeros (atwos, alow);
  116.   TRACE (printf ("a twos %un", atwos));
  117.   result_bit1 ^= JACOBI_TWOS_U_BIT1 (atwos, blow);
  118.   /* establish shifted alow */
  119.   alow >>= atwos;
  120.   if (asize > 1)
  121.     {
  122.       asecond = asrcp[1];
  123.       if (atwos != 0)
  124.         alow |= (asecond << (GMP_NUMB_BITS - atwos)) & GMP_NUMB_MASK;
  125.     }
  126.   /* (a/2)=(2/a) with a odd */
  127.   result_bit1 ^= JACOBI_TWOS_U_BIT1 (btwos, alow);
  128.   if (asize == 1 || (asize == 2 && (asecond >> atwos) == 0))
  129.     {
  130.       /* another special case with modexact and no copying */
  131.       if (alow == 1)  /* (1/b)=1 always */
  132.         return JACOBI_BIT1_TO_PN (result_bit1);
  133.       /* b still has its twos, so cancel out their effect */
  134.       result_bit1 ^= JACOBI_TWOS_U_BIT1 (btwos, alow);
  135.       result_bit1 ^= JACOBI_RECIP_UU_BIT1 (alow, blow);  /* now (b/a) */
  136.       JACOBI_MOD_OR_MODEXACT_1_ODD (result_bit1, blow, bsrcp, bsize, alow);
  137.       TRACE (printf ("base (%lu/%lu) with %dn",
  138.                      blow, alow, JACOBI_BIT1_TO_PN (result_bit1)));
  139.       return mpn_jacobi_base (blow, alow, result_bit1);
  140.     }
  141.   TMP_MARK;
  142.   TMP_ALLOC_LIMBS_2 (ap, asize, bp, bsize);
  143.   MPN_RSHIFT_OR_COPY (ap, asrcp, asize, atwos);
  144.   ASSERT (alow == ap[0]);
  145.   TRACE (mpn_trace ("stripped a", ap, asize));
  146.   MPN_RSHIFT_OR_COPY (bp, bsrcp, bsize, btwos);
  147.   ASSERT (blow == bp[0]);
  148.   TRACE (mpn_trace ("stripped b", bp, bsize));
  149.   /* swap if necessary to make a longer than b */
  150.   if (asize < bsize)
  151.     {
  152.       TRACE (printf ("swapn"));
  153.       MPN_PTR_SWAP (ap,asize, bp,bsize);
  154.       MP_LIMB_T_SWAP (alow, blow);
  155.       result_bit1 ^= JACOBI_RECIP_UU_BIT1 (alow, blow);
  156.     }
  157.   /* If a is bigger than b then reduce to a mod b.
  158.      Division is much faster than chipping away at "a" bit-by-bit. */
  159.   if (asize > bsize)
  160.     {
  161.       mp_ptr  rp, qp;
  162.       TRACE (printf ("tdiv_qr asize=%ld bsize=%ldn", asize, bsize));
  163.       TMP_ALLOC_LIMBS_2 (rp, bsize, qp, asize-bsize+1);
  164.       mpn_tdiv_qr (qp, rp, (mp_size_t) 0, ap, asize, bp, bsize);
  165.       ap = rp;
  166.       asize = bsize;
  167.       MPN_NORMALIZE (ap, asize);
  168.       TRACE (printf ("tdiv_qr asize=%ld bsize=%ldn", asize, bsize);
  169.              mpn_trace (" a", ap, asize);
  170.              mpn_trace (" b", bp, bsize));
  171.       if (asize == 0)  /* (0/b)=0 for b!=1 */
  172.         goto zero;
  173.       alow = ap[0];
  174.       goto strip_a;
  175.     }
  176.   for (;;)
  177.     {
  178.       ASSERT (asize >= 1);         /* a,b non-empty */
  179.       ASSERT (bsize >= 1);
  180.       ASSERT (ap[asize-1] != 0);   /* a,b normalized (and hence non-zero) */
  181.       ASSERT (bp[bsize-1] != 0);
  182.       ASSERT (alow == ap[0]);      /* low limb copies should be correct */
  183.       ASSERT (blow == bp[0]);
  184.       ASSERT (alow & 1);           /* a,b odd */
  185.       ASSERT (blow & 1);
  186.       TRACE (printf ("top asize=%ld bsize=%ldn", asize, bsize);
  187.              mpn_trace (" a", ap, asize);
  188.              mpn_trace (" b", bp, bsize));
  189.       /* swap if necessary to make a>=b, applying reciprocity
  190.          high limbs are almost always enough to tell which is bigger */
  191.       if (asize < bsize
  192.           || (asize == bsize
  193.               && ((ahigh=ap[asize-1]) < (bhigh=bp[asize-1])
  194.                   || (ahigh == bhigh
  195.                       && mpn_cmp (ap, bp, asize-1) < 0))))
  196.         {
  197.           TRACE (printf ("swapn"));
  198.           MPN_PTR_SWAP (ap,asize, bp,bsize);
  199.           MP_LIMB_T_SWAP (alow, blow);
  200.           result_bit1 ^= JACOBI_RECIP_UU_BIT1 (alow, blow);
  201.         }
  202.       if (asize == 1)
  203.         break;
  204.       /* a = a-b */
  205.       ASSERT (asize >= bsize);
  206.       ASSERT_NOCARRY (mpn_sub (ap, ap, asize, bp, bsize));
  207.       MPN_NORMALIZE (ap, asize);
  208.       alow = ap[0];
  209.       /* (0/b)=0 for b!=1.  b!=1 when a==0 because otherwise would have had
  210.          a==1 which is asize==1 and would have exited above.  */
  211.       if (asize == 0)
  212.         goto zero;
  213.     strip_a:
  214.       /* low zero limbs on a can be discarded */
  215.       JACOBI_STRIP_LOW_ZEROS (result_bit1, blow, ap, asize, alow);
  216.       if ((alow & 1) == 0)
  217.         {
  218.           /* factors of 2 from a */
  219.           unsigned  twos;
  220.           count_trailing_zeros (twos, alow);
  221.           TRACE (printf ("twos %un", twos));
  222.           result_bit1 ^= JACOBI_TWOS_U_BIT1 (twos, blow);
  223.           ASSERT_NOCARRY (mpn_rshift (ap, ap, asize, twos));
  224.           asize -= (ap[asize-1] == 0);
  225.           alow = ap[0];
  226.         }
  227.     }
  228.   ASSERT (asize == 1 && bsize == 1);  /* just alow and blow left */
  229.   TMP_FREE;
  230.   /* (1/b)=1 always (in this case have b==1 because a>=b) */
  231.   if (alow == 1)
  232.     return JACOBI_BIT1_TO_PN (result_bit1);
  233.   /* swap with reciprocity and do (b/a) */
  234.   result_bit1 ^= JACOBI_RECIP_UU_BIT1 (alow, blow);
  235.   TRACE (printf ("base (%lu/%lu) with %dn",
  236.                  blow, alow, JACOBI_BIT1_TO_PN (result_bit1)));
  237.   return mpn_jacobi_base (blow, alow, result_bit1);
  238.  zero:
  239.   TMP_FREE;
  240.   return 0;
  241. }