【DP】【P1586】四方定理

传送门

Description

Input

第一行为一个整数T代表数据组数，之后T行每行一个数n代表要被分解的数

Output

对于每个n输出一行，为方案个数

Sample Input

Sample Output

Hint

t<=100,n<=23768。

Solution

　　dp方程转移之类显然，唯一需要说的是有关去重的问题。显然需要打一张到maxn的平方表。然后f[i][j]代表i分解为j个平方数的方案数。如题面所说，x=a²+b²与x=b²⁺a²是同一种方案。既然如此，就不能外层循环第一维度内层循环平方表进行转移，因为这样在枚举了第一种方案后会继续枚举相同的第二种方案。考虑阶段：对于一个数x，它的分解方式可分为两个类，第一种是包含a²的，第二种是不包含a²的。对于这两个类分别枚举转移，则不会产生重复。正确性显然。故正确的枚举顺序是外层为平方表，内层是数字x，第三层是分解个数k。

Code

#include<cstdio>
#define maxn 40000
#define maxk 32769

inline void qr(int &x) {
    char ch=getchar();int f=;
    while(ch>''||ch<'')    {
        if(ch=='-')    f=-;
        ch=getchar();
    }
    while(ch>=''&&ch<='')    x=(x<<)+(x<<)+(ch^),ch=getchar();
    x*=f;
    return;
}

inline int max(const int a,const int b) {if(a>b) return a;else return b;}
inline int min(const int a,const int b) {if(a<b) return a;else return b;}
inline int abs(const int x) {if(x>) return x;else return -x;}

inline void swap(int &a,int &b) {
    int c=a;a=b;b=c;return;
}

const int biao[]={,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,,};
int t,frog[maxn][],a;

inline int sigma(int a) {return frog[a][]+frog[a][]+frog[a][]+frog[a][];}

int main() {
    qr(t);
    frog[][]=;
    for(int i=;i<=;++i) {
        for(int j=biao[i];j^maxk;++j) {
            for(int k=;k^;++k) frog[j][k]+=frog[j-biao[i]][k-];
        }
    }
    do {
        a=;qr(a);printf("%d\n",sigma(a));
    } while(--t);
    return ;
}

Summary

　　对于需要去重的方案个数DP，可以考虑将一个状态分解为包括一个元素的状态和不包括一个元素的状态，以被包括的元素做阶段进行转移，则不存在重复问题。