洛谷 P1586 四方定理

题目传送门

解题思路:

f[i][j]表示数i被j个数构成的方案数,然后就跟完全背包差不多了

AC代码:

#include<iostream>
#include<cstdio>

using namespace std;

int t,n = 32768,f[32800][5];

int main() {
    f[0][0] = 1;
    for(int i = 1;i * i <= n; i++)
        for(int j = i * i;j <= n; j++)
            for(int k = 1;k <= 4; k++)
                f[j][k] += f[j-i*i][k-1];
    scanf("%d",&t);
    while(t--) {
        scanf("%d",&n);
        int ans = 0;
        for(int i = 1;i <= 4; i++)
            ans += f[n][i];
        printf("%d
",ans);
    }
    return 0;
}