洛谷P1586 四方定理 动态规划 + 离线

洛谷P1586 四方定理
动态规划 + 离线

#include <bits/stdc++.h> 
#define For(i,j,k) for(int i=j;i<=k;i++) 
using namespace std ; 

const int N = 35011 ; 
int n,T,ans ; 
int f[5][N],a[211] ; 

inline 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-48 ; ch = getchar() ; } 
    return x * f ;  
}

inline void write(int x) 
{
    if(x<0) x=-x,putchar('-') ; 
    if(x>9) write(x/10) ; 
    putchar(x%10+48) ; 
}

inline void writeln(int x) 
{
    write(x) ;  putchar('
') ; 
}

int main() 
{
    T = read() ;
    For(i,1,200) a[ i ] = i*i ;  f[0][0] = 1 ; 
    For(i,1,200) 
      For(k,1,4)
        For(j,a[i],32769) 
            f[k][j]+=f[k-1][j-a[i]] ;       
    while(T--) {
        n = read() ;
        ans = 0 ;  
        For(i,1,4) ans+=f[i][n] ; 
        writeln(ans) ; 
    }
    return 0 ; 
}