P1586 四方定理

題目描述

四方定理是眾所周知的：任意一個正整數nn，可以分解為不超過四個整數的平方和。例如：25=1^{2}+2^{2}+2^{2}+4^{2}25=12+22+22+42，當然還有其他的分解方案，25=4^{2}+3^{2}25=42+32和25=5^{2}25=52。給定的正整數nn，編程統計它能分解的方案總數。註意：25=4^{2}+3^{2}25=42+32和25=3^{2}+4^{2}25=32+42視為一種方案。

輸入輸出格式

輸入格式：

第一行為正整數tt(t\le 100t≤100)，接下來tt行，每行一個正整數nn(n\le 32768n≤32768)。

輸出格式：

對於每個正整數nn，輸出方案總數。

輸入輸出樣例

1
2003

48

枚舉

當前數最多由4個四方數組成，那麽我們可以枚舉這4個數，然後循環枚舉，我們可以直接處理到最大的數，這樣就可以不用處理t次了

然後我們再在美劇裏面加一點剪枝就好了

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define n 182 
#define
 
 N 40000
using namespace std;
int m,t,a[n],ans[N],f[n],maxn;
int read()
{
    int x=0,f=1; char ch=getchar();
    while(ch<‘0‘||ch>‘9‘){if(ch==‘-‘)f=-1;ch=getchar();}
    while(ch>=‘0‘&&ch<=‘9‘) x=x*10+ch-‘0‘,ch=getchar();
    return x*f;
}
int main()
{
    t=read();
    for(int
 
 i=1;i<=t;i++)
     a[i]=read(),maxn=max(maxn,a[i]);
    for(int i=1;i<=n;i++)
     f[i]=i*i;
    for(int i=0;i<=n;i++)
    {
        if(f[i]>maxn) break;
        for(int j=i;j<=n;j++)
        {
            if(f[i]+f[j]>maxn) break;
            for(int x=j;x<=n;x++)
            {
                if(f[i]+f[j]+f[x]>maxn) break;
                for(int y=x;y<=n;y++)
                 if(f[i]+f[j]+f[x]+f[y]>maxn) break;
                 else ans[f[i]+f[j]+f[x]+f[y]]++;
            }
        }
    }
    for(int i=1;i<=t;i++) 
     printf("%d\n",ans[a[i]]);
    return 0;
}

洛谷——P1586 四方定理