1. 程式人生 > >快速pow和sqrt的小技巧 hdu4282

快速pow和sqrt的小技巧 hdu4282

今年網路賽。。天津賽區。。有道題。。是這樣的。。。X^Z + Y^Z + XYZ = K  給出K ,求XYZ,我思路很明確。。。列舉其二,然後二分其一,但是始終TLE。。。。晚上回去之後,看了人家報告,。。。才發現。。。原來是微軟的函式pow 惹的禍。。。我本來以為微軟的函式寫的都很好。。效率很高。。但是我忘了一件事。。就是 pow(int a, int b) 這個函式可以理解為 pow(double a, double b)  之所以TLE  。。是因為微軟提高的相容性。。而導致時間效率的底下。。(但是令我鬱悶的就是。。網上有位仁兄。。也用的pow。。居然984ms 壓線過了這道題。。。汗!= =||)。。所以我就百度了一下。。找到了快速pow 和sqrt的方法~~

如下:

  1. **=============================================**  
  2. ||快速pow(多次使用時及其有效)               ||  
  3. **=============================================**  
  4. __int64 qpow(int a, int b){  
  5.     __int64 c, d; c = 1; d = a;  
  6.     while (b > 0){  
  7.         if (b & 1)  
  8.           c *= d;  
  9.         b = b >> 1;  
  10.         d = d * d;  
  11.     }  
  12.     return c;  
  13. }  
  14. **=============================================**  
  15. ||快速1/sqrt(x)(牛頓迭代法)                    ||  
  16. **=============================================**  
  17. float InvSqrt (float x) {  
  18.      float xhalf = 0.5f * x;  
  19.      long i = *(long*)&x;  
  20.      i = 0x5f3759df - (i >> 1);  
  21.      x = *(float
     *)&i;  
  22.      x = x * (1.5f - xhalf * x * x);  
  23.      return x;  
  24. }  

再列出我最後AC的程式碼:

  1. #include <stdio.h>  
  2. #include <math.h>  
  3. __int64 k;  
  4. //其實我也試著寫了一個pow。。只不過弱爆了。。不夠快。。  
  5. /* 
  6. __int64 myqpow(__int64 x, __int64 y) 
  7. { 
  8.     if (y == 1) 
  9.     { 
  10.         return x; 
  11.     } 
  12.     __int64 tmp = qpow(x, y/2); 
  13.     if (y%2 == 1) 
  14.     { 
  15.         return (tmp * tmp * x); 
  16.     } 
  17.     else 
  18.     { 
  19.         return (tmp * tmp); 
  20.     } 
  21. } 
  22. */  
  23. __int64 qpow(int a, int b)  
  24. {  
  25.     __int64 c, d; c = 1; d = a;  
  26.     while (b > 0)  
  27.     {  
  28.         if (b & 1)  
  29.           c *= d;  
  30.         b = b >> 1;  
  31.         d = d * d;  
  32.     }  
  33.     return c;  
  34. }  
  35. __int64 solve(__int64 x, __int64 z)  
  36. {  
  37.     __int64 l = x + 1, r = 32768, y = (l + r) >> 1;//r = k - qpow(x, z)  
  38.     while (l <= r)  
  39.     {  
  40.         __int64 tmp = x * y * z + qpow(x, z) + qpow(y, z);  
  41.         if (tmp == k)  
  42.         {  
  43.             return y ;  
  44.         }  
  45.         else if (tmp > k || tmp < 0)  
  46.         {  
  47.             r = y - 1;  
  48.         }  
  49.         else  
  50.         {  
  51.             l = y + 1;  
  52.         }  
  53.         y = (l + r) >> 1;  
  54.     }  
  55. /* 
  56.     if (x * y * z + qpow(x, z) + qpow(y, z) == k) 
  57.     { 
  58.         return y ; 
  59.     } 
  60. */  
  61.     return 0 ;  
  62. }  
  63. int main()  
  64. {  
  65.     while (scanf("%I64d", &k), k)      
  66.     {  
  67.         __int64 i, j, ans=0;  
  68.         for (i = 2; i <= 31; i++)  
  69.         {  
  70.             for (j = 1;  ; j++)  
  71.             {  
  72.                 __int64 tmp1 = qpow(j, i);  
  73.                 if (tmp1*2 > k || tmp1 < 0)  
  74.                 {  
  75.                     break ;  
  76.                 }  
  77. //                __int64 a = solve(j, i);  
  78.                 if (solve(j, i))  
  79.                 {  
  80.                     ans++;  
  81. //                    printf("%I64d %I64d %I64d\n", j, a, i);  
  82.                 }  
  83.             }  
  84.         }  
  85.         printf("%I64d\n", ans);  
  86.     }  
  87.     return 0;  
  88. }  
  89. //  5 6 5  
  90. //  3125 7776 150  
  91. //  11051  
  92. //  40 48 5  
  93. //  102400000  254803968 9600  
  94. //  357213568