以下代码是确定整数三角形数量的算法,最大边小于或等于 MAX,具有整数中位数。Python 版本可以工作,但对于更大的 N 来说太慢了,而 C++ 版本要快得多,但没有给出正确的结果。
当 MAX 为 10 时,C++ 和 Python 都返回 3。
当 MAX 为 100 时,Python 返回 835,C++ 返回 836。
当 MAX 为 200 时,Python 返回 4088,C++ 返回 4102。
当 MAX 为 500 时,Python 返回 32251,C++ 返回 32296。
当 MAX 为 1000 时,Python 返回 149869,C++ 返回 150002。
这是 C++ 版本:
#include <cstdio>
#include <math.h>
const int MAX = 1000;
int main()
{
long long int x = 0;
for (int b = MAX; b > 4; b--)
{
printf("%lld\n", b);
for (int a = b; a > 4; a -= 2){
for (int c = floor(b/2); c < floor(MAX/2); c+=1)
{
if (a+b > 2*c){
int d = 2*(pow(a,2)+pow(b,2)-2*pow(c,2));
if (sqrt(d)/2==floor(sqrt(d)/2))
x+=1;
}
}
}
}
printf("Done: ");
printf("%lld\n", x);
}
这是原始的 Python 版本:
import math
def sumofSquares(n):
f = 0
for b in range(n,4,-1):
print(b)
for a in range(b,4,-2):
for C in range(math.ceil(b/2),n//2+1):
if a+b>2*C:
D = 2*(a**2+b**2-2*C**2)
if (math.sqrt(D)/2).is_integer():
f += 1
return f
a = int(input())
print(sumofSquares(a))
print('Done')
我对 C++ 不太熟悉,所以我不知道会发生什么导致这种情况(可能是溢出错误?)。
当然,任何对算法的优化都非常受欢迎!