3234801_wa.cc

来自「北大大牛代码 1240道题的原代码 超级权威」· CC 代码 · 共 20 行

#include <stdio.h>

int num[] = {0,2,3,9,11};

int main()
{
	int cas;
	__int64 n;

	scanf("%d",&cas);
	while(cas-- > 0)
	{
		scanf("%I64d",&n);
		if(n < 5)
			printf("%d\n",num[n]);
		else
			printf("%I64d\n",n*n-1);
	}
	return 0;
}