1387: [蓝桥杯2018决赛]阅兵方阵

这道题说白了就是把一个数分成两个平方数，问这一组平方数有多少种，注意：

例如：
1、9 = 1 + 8
2、9 = 8 + 1
∵ 上述两种情况其实是同一个队形
∴ 要求变换的队形数量要 * 2

#include<bits/stdc++.h>
using namespace std;
#define ll long long
int q[105];
int w[105];

int main()
{

    for(int i=1; ;i++)
    {
        int cnt=0;
        for(int j=1; j*j<=i; j++)
        {
            int k=sqrt(i-j*j);
            if(k*k+j*j==i)
                cnt++;
            if(cnt==24)
            {
                printf("%d\n",i);
                return 0;
            }
        }

    }
//    cout<<160225<<endl;

}