zoj 1738 Lagrange's Four-Square Theorem

/*zoj_1738 dp思路：状态方程dp[i][j]表示元素i分解成j个数的平方和的可能数*/#include <iostream>#include <cstdio>#include <string.h>#define N 40000using 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;}