寻找100到1000之间的所有毕达哥拉斯三胞胎

时间:2016-01-19 10:15:16

标签: c++ c++11 cmath pythagorean triplet

我写了一个程序,找到100到1000之间的毕达哥拉斯三胞胎。 这里的代码是相同的。

#include<iostream>
#include<cmath>

using namespace std;

bool checkWhetherInteger(int x, int y);
int getInteger(int x, int y);

int main() 
{
    cout << "This program is to print all pythagorean triplets from 100 to 1000. \n";

    int x=100;
    int y=100;

    for(; x<=1000; x++)
    {
        for (; y<=1000; y++)
        {
            if (checkWhetherInteger(x,y))
            {
                cout << "Triplet : " << x << "  " << y << "  " << getInteger(x,y) << "\n";
            }
        }
        y=100;
    }
    return 0;
}

bool checkWhetherInteger(int x, int y)
{
    for (int i=141; i<=1415; i++)
    {
        if(hypot(x,y) == i )
        {
            return true;
        }
    }
}

int getInteger(int x, int y)
{
    return static_cast<int>(hypot(x,y));

}

我现在面临两个问题。

  1. 主要问题是执行缓慢。虽然它给了我所有三胞胎,但执行时间约为553.7秒。那么有什么算法可以让我想要的东西在1-2秒或更短的时间内完成。

  2. 由于两个独立的变量x和y,我得到一个三联体两次。可以做些什么。

  3. 如果我犯了一些错误,请耐心等待。我是学习者。

1 个答案:

答案 0 :(得分:1)

查找所有三元组的最快方法是使用此公式:

a = k(x^2 - y^2)
b = k(2xy)
c = k(x^2 + y^2)

https://en.wikipedia.org/wiki/Pythagorean_triple

删除重复项的标准(以及最快速的方法之一):使用std :: set&lt;&gt;。

代码示例:

#include <iostream>
#include <string>
#include <vector>
#include <set>

struct Triple {
  int a;
  int b;
  int c;
};

struct comp {
  inline bool operator ()(const Triple & first, const Triple & second) const {
    if (first.a != second.a) {
      return first.a < second.a;
    }
    if (first.b != second.b) {
      return first.b < second.b;
    }
    return first.c < second.c;
  }
};

int main() {
  int n = 1000;
  std::set<Triple, comp> set;
  for (int x = 2; x <= n; ++x) {
    for (int y = 1; y < x && x * x + y * y <= n; ++y) {
      int a = x * x - y * y;
      int b = 2 * x * y;
      if (a > b) {
        std::swap(a, b);
      }
      int c = x * x + y * y;
      for (int k = 1; k * c <= n; ++k) {
        if (a * k >= 100 && a * k <= n &&
            b * k >= 100 && b * k <= n &&
            c * k >= 100 && c * k <= n)
        set.insert({k * a, k * b, k * c});
      }
    }
  }
  for (const auto & triple : set) {
    std::cout << triple.a << " " << triple.b << " " << triple.c << std::endl;
  }
  return 0;
}

coliru