P1586 四方定理

題目描述

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

輸入輸出格式

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

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

輸入輸出樣例

1
2003

48
思路：1.四重循環。
錯因：輸出沒有換行。

#include<iostream>
#include<cmath>
#include<cstdio>
#include<cstring>
using namespace std;
int t,n,ans;
int main(){
    cin>>t;
    while(t--){
        cin>>n;
        ans
 
=0;
        for(int ii=0;ii*ii<=n;ii++)
            for(int j=ii;ii*ii+j*j<=n;j++)
                for(int k=j;k*k+j*j+ii*ii<=n;k++){
                    int num=n-ii*ii-j*j-k*k;
                    int s=(int)sqrt(num);
                    if(s*s==num&&k<=s)
                        ans
 
++;
                }
        cout<<ans<<endl;
    }
}

思路：2.dp可以列出狀態轉移方程f[i][j]=Σf[i-k*k][j-1];

#include<iostream>
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int t,n,f[100000][5];
int main(){
    f[0][0]=1;
    for(int k=1;k*k<=32768;k++)
        for(int i=k*k;i<=32768;i++)
            for(int j=1;j<=4;j++)
                f[i][j]+=f[i-k*k][j-1];
    cin>>t;
    while(t--){
        cin>>n;
        cout<<f[n][1]+f[n][2]+f[n][3]+f[n][4]<<endl;
    }
}

洛谷 P1586 四方定理