【Luogu】P1586四方定理（DP）

题目链接

此题使用DP。设f[i][j]表示数i用j个数表示，则对于所有的k<=sqrt(i),有

f[i][j]=∑f[i-k*k][j-1]

但是这样会有重复情况。所以先枚举k，再枚举i和j。

代码如下

#include<cstdio>
#include<cmath>
#include<cstdlib>
#include<cstring>
#include<cctype>
inline long long read(){
    long long num=,f=;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')    f=-;
        ch=getchar();
    }
    while(isdigit(ch)){
        num=num*+ch-'';
        ch=getchar();
    }
    return num*f;
}

long long f[][]={};
int main(){
    int T=read();
    for(int k=;k*k<=;++k)
        for(int i=k*k;i<=;++i)
            for(int j=;j<=;++j)
                f[i][j]+=f[i-k*k][j-];
    while(T--){
        int n=read();
        printf("%lld\n",f[n][]+f[n][]+f[n][]+f[n][]);
    }
    return ;
}