四方和问题。求更简单的代码,我这个会超时,(测试数据773635),谢谢
#include <iostream>#include <cmath>
using namespace std;
int main()
{
long long N, i, j, p, q;
while(cin >> N)
{
int flag = 0;
long long n = sqrt(N);
for(i = 0; i <= n; i++)
{
for(j = i; j <= n; j++)
{
long long s = N-i*i-j*j;
for(p = j; p < s; p++)
{
for(q = p; q <= s&&q>=p; q++)
{
if(s==p*p+q*q)
{
cout << i << " " << j <<" "<< p <<" "<< q <<endl;
flag = 1;
break;
}
}
if(flag) break;
}
if(flag) break;
}
if(flag) break;
}
}
return 0;
}