智慧数

《关于当代大学生被一道初中数学题难住所得灵感写的一道Java版智慧数》题目是这样的：

一个正整数如果能表示成了两个正整数的平方差，则称这个数为“智慧数”，比如16就等于5的平方减去3的平方，所以16就是一个智慧数，从1开始的自然数列中，将“智慧数”从小到大编号为1,2,3，„„，n。现输入一个正整数n，求第2006个“智慧数”。

输入

3

输出

7

输入4

输出8

这道题我是根据纯数学理论所做，如果有大佬有更好的方法，请在评论里指点我一下。

	//函数体
    public int number(int n){
        if(n<4){
            if(n==1){
                return 3;
            }else if(n==2){
                return 5;
            }else if(n==3){
                return 7;
            }else {
                return 0;
            }
        }else {
            if(n%3==0){
                int num =(int) 4*(n/3)+3;
                return num;
            }else if(n%3==1){
                int num = (int) 4*((n/3)+1);
                return num;
            }else {
                int num = (int) 4*((n/3)+1)+1;
                return num;
            }
        }
    }