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

数学计算

开发平台:

Unix_Linux

  1. /* Test mpz_root, mpz_rootrem, and mpz_perfect_power_p.
  2. Copyright 1991, 1993, 1994, 1996, 2000, 2001, 2009 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
  5. it under the terms of the GNU Lesser General Public License as published by
  6. the 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
  10. or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public
  11. License for more details.
  12. You should have received a copy of the GNU Lesser General Public License
  13. along with the GNU MP Library.  If not, see http://www.gnu.org/licenses/.  */
  14. #include <stdio.h>
  15. #include <stdlib.h>
  16. #include "gmp.h"
  17. #include "gmp-impl.h"
  18. #include "tests.h"
  19. void debug_mp __GMP_PROTO ((mpz_t, int));
  20. void
  21. check_one (mpz_t root1, mpz_t x2, unsigned long nth, int i)
  22. {
  23.   mpz_t temp, temp2;
  24.   mpz_t root2, rem2;
  25.   mpz_init (root2);
  26.   mpz_init (rem2);
  27.   mpz_init (temp);
  28.   mpz_init (temp2);
  29.   MPZ_CHECK_FORMAT (root1);
  30.   mpz_rootrem (root2, rem2, x2, nth);
  31.   MPZ_CHECK_FORMAT (root2);
  32.   MPZ_CHECK_FORMAT (rem2);
  33.   mpz_pow_ui (temp, root1, nth);
  34.   MPZ_CHECK_FORMAT (temp);
  35.   mpz_add (temp2, temp, rem2);
  36.   /* Is power of result > argument?  */
  37.   if (mpz_cmp (root1, root2) != 0 || mpz_cmp (x2, temp2) != 0 || mpz_cmp (temp, x2) > 0)
  38.     {
  39.       fprintf (stderr, "ERROR after test %dn", i);
  40.       debug_mp (x2, 10);
  41.       debug_mp (root1, 10);
  42.       debug_mp (root2, 10);
  43.       fprintf (stderr, "nth: %lun", nth);
  44.       abort ();
  45.     }
  46.   if (nth > 1 && mpz_cmp_ui (temp, 1L) > 0 && ! mpz_perfect_power_p (temp))
  47.     {
  48.       fprintf (stderr, "ERROR in mpz_perfect_power_p after test %dn", i);
  49.       debug_mp (temp, 10);
  50.       debug_mp (root1, 10);
  51.       fprintf (stderr, "nth: %lun", nth);
  52.       abort ();
  53.     }
  54.   if (nth <= 10000) /* skip too expensive test */
  55.     {
  56.       mpz_add_ui (temp2, root1, 1L);
  57.       mpz_pow_ui (temp2, temp2, nth);
  58.       MPZ_CHECK_FORMAT (temp2);
  59.       /* Is square of (result + 1) <= argument?  */
  60.       if (mpz_cmp (temp2, x2) <= 0)
  61. {
  62.   fprintf (stderr, "ERROR after test %dn", i);
  63.   debug_mp (x2, 10);
  64.   debug_mp (root1, 10);
  65.   fprintf (stderr, "nth: %lun", nth);
  66.   abort ();
  67. }
  68.     }
  69.   mpz_clear (root2);
  70.   mpz_clear (rem2);
  71.   mpz_clear (temp);
  72.   mpz_clear (temp2);
  73. }
  74. int
  75. main (int argc, char **argv)
  76. {
  77.   mpz_t x2;
  78.   mpz_t root1;
  79.   mp_size_t x2_size;
  80.   int i;
  81.   int reps = 500;
  82.   unsigned long nth;
  83.   gmp_randstate_ptr rands;
  84.   mpz_t bs;
  85.   unsigned long bsi, size_range;
  86.   tests_start ();
  87.   TESTS_REPS (reps, argv, argc);
  88.   rands = RANDS;
  89.   mpz_init (bs);
  90.   mpz_init (x2);
  91.   mpz_init (root1);
  92.   /* This triggers a gcc 4.3.2 bug */
  93.   mpz_set_str (x2, "ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff80000000000000000000000000000000000000000000000000000000000000002", 16);
  94.   mpz_root (root1, x2, 2);
  95.   check_one (root1, x2, 2, -1);
  96.   for (i = 0; i < reps; i++)
  97.     {
  98.       mpz_urandomb (bs, rands, 32);
  99.       size_range = mpz_get_ui (bs) % 17 + 2;
  100.       mpz_urandomb (bs, rands, size_range);
  101.       x2_size = mpz_get_ui (bs) + 10;
  102.       mpz_rrandomb (x2, rands, x2_size);
  103.       mpz_urandomb (bs, rands, 15);
  104.       nth = mpz_getlimbn (bs, 0) % mpz_sizeinbase (x2, 2) + 2;
  105.       mpz_root (root1, x2, nth);
  106.       mpz_urandomb (bs, rands, 4);
  107.       bsi = mpz_get_ui (bs);
  108.       if ((bsi & 1) != 0)
  109. {
  110.   /* With 50% probability, set x2 near a perfect power.  */
  111.   mpz_pow_ui (x2, root1, nth);
  112.   if ((bsi & 2) != 0)
  113.     {
  114.       mpz_sub_ui (x2, x2, bsi >> 2);
  115.       mpz_abs (x2, x2);
  116.     }
  117.   else
  118.     mpz_add_ui (x2, x2, bsi >> 2);
  119.   mpz_root (root1, x2, nth);
  120. }
  121.       check_one (root1, x2, nth, i);
  122.     }
  123.   mpz_clear (bs);
  124.   mpz_clear (x2);
  125.   mpz_clear (root1);
  126.   tests_end ();
  127.   exit (0);
  128. }
  129. void
  130. debug_mp (mpz_t x, int base)
  131. {
  132.   mpz_out_str (stderr, base, x); fputc ('n', stderr);
  133. }