四方定理是众所周知的:任意一个正整数n,可以分解为不超过四个整数的平方和
给定的正整数n,统计它能分解的方案总数。注意:25=4^2+3^2 与 3^2+4^2 相同
背包,把每个平方数作为一个物品
直接用背包会重复累加,此时要利用这个定理,f[4][j]
#include <iostream>
#include <cstring>
using namespace std;
const int N=32769;
int f[5][N+5],b[N+5];
void init(){
int i,j,k;
for(int i=1;i*i<=N;i++) b[i*i]=1;
memset(f,0,sizeof f);
f[0][0]=1;
for(i=1;i<=N;i++) if(b[i])
for(j=i;j<=N;j++)
for(k=1;k<=4;k++)
f[k][j]+=f[k-1][j-i];
}
signed main(){
init();
int cas,x;
cin>>cas;
while(cas--){
int s=0;
cin>>x;
for(int k=1;k<=4;k++) s+=f[k][x];
cout<<s<<endl;
}
}