zoj 1738 Lagrange's Four-Square Theorem

/*
zoj_1738 dp
思路：状态方程dp[i][j]表示元素i分解成j个数的平方和的可能数
*/
#include <iostream>
#include <cstdio>
#include <string.h>
#define N 40000
using namespace std;
int dp[N][5];

int main()
{
    int n,i,j,k;
    memset( dp,0,sizeof(dp) );
    dp[0][0]=1;
    for( i=1;i*i<N;i++ )
        for( j=1;j<5;j++ )
            for( k=i*i;k<N;k++ )
                dp[k][j]+=dp[k-i*i][j-1];
    while( cin>>n && n )
        printf( "%d\n",dp[n][1]+dp[n][2]+dp[n][3]+dp[n][4] );
    return 0;
}