华为笔试：直角三角形个数

题目描述：

给定三角形周长p，求满足边长为整数且周长为p的直角三角形个数。

 

思路分析：

枚举的思想。首先想到就是利用一个双重循环：

for(int i=1; i<p; i++)
{
    for(int j=i; j<p; j++)
    {
        int k = p-i-j;
        if(i*i+j*j==k*k)
            ans++;
    }
}

但这样是会超时的，通过数学方式做分析：

i+j+k=p, 0<i<=j<k, 通过解不等式，可以得到：i<p/3, j<p/2。

在双重循环的基础上，做限制，能够通过，但仍可以进一步优化。

 

一重循环：

考虑两个方程：

i+j+k=p, i^2+j^2 = k^2。

j = p-p^2/(2p-2i)。

 

代码：

#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;

int main(){
    int p;
    cin>>p;
    int ans=0;
    for(int i=1; i<p/3; i++)
    {
        double j=p-(double)p*p/(2*p-2*i);
        if(i<j && j-(int)j<1e-5)
        {
            ans++;
        }
    }
    cout<<ans<<endl;
    return 0;
    
}