Find Integer - hdu6441 - 费马大定理+奇偶数列法则

题目入口

题目给定了a,则:
若n>2由费马大定理知无解。输出-1 -1.
n=0时也无解，输出-1 -1
若n=1，则a+b=c，输出1 a+1
若n=2，则a^ 2+b^ 2=c^2，由奇偶数列法则得到：

若a为奇数:

a=2k+1 （k=1,2,3…）
b= k^ 2+(k+1)^2-1
c= k^ 2+(k+1)^2

若a为偶数:

a=2k（k=1,2,3…）
b= k^2 -1
c= k^2 +1

详细证明:直角三角形a^ 2+b^ 2=c^2 整数解的定a公式直求法
(一个小姐姐的博客)

#include<bits/stdc++.h>
using namespace std;

int main()
{
	int T;
	scanf("%d",&T);
	while(T--){
		int n,a;
		scanf("%d%d",&n,&a);
		if(n>2||n==0) printf("-1 -1\n");
		else if(n==1) printf("1 %d\n",a+1);
		else if(n==2){
			if(a%2==0){
				a = a / 2;
				printf("%d %d\n",a*a-1,a*a+1);
			}
			else {
				a = (a-1)/2;
				printf("%d %d\n",a*a + (a+1)*(a+1) -1, a*a+(a+1)*(a+1));
			}
		}
	}
	return 0;
}