0

以下代码是确定整数三角形数量的算法,最大边小于或等于 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++ 不太熟悉,所以我不知道会发生什么导致这种情况(可能是溢出错误?)。

当然,任何对算法的优化都非常受欢迎!

4

2 回答 2

1

问题是您的cC在 python 中)变量的范围不匹配。为了使它们等同于您现有的 C++ 范围,您可以将您的 python 循环更改为:

for C in range(int(math.floor(b/2)), int(math.floor(n/2))):
    ...

为了使它们等同于您现有的 python 范围,您可以将 C++ 循环更改为:

for (int c = ceil(b/2.0); c < MAX/2 + 1; c++) {
    ...
}

根据最初是正确的循环,这将使结果匹配。

于 2015-04-28T04:46:47.213 回答
-1

它接缝了一些麻烦可能在这里:

(sqrt(d)==floor(sqrt(d))) 
于 2015-04-28T04:38:13.490 回答