_xiaobai_

导航

zoj1738 Lagrange's Four-Square Theorem(DP)

/*
 经典模型:完全背包
 本问题有两种考虑思路:
 1.一维多重背包,每个阶段由一个状态改为四个状态,容易想到跟阶段间的转移方程。
 2.二维完全背包,同一维一样,转移方程:
  F(i,j,k) = opt{ F(i-V(i),j-C(i),k-1),F(i,j,k-1) } (k为阶段)
*/

View Code
 1 #include <stdio.h>
2 #include <string.h>
3
4 int F[ 5 ][ 32770 ];
5
6 int main()
7 {
8 int i,j,k;
9 memset( F, 0, sizeof( F ) );
10 F[ 0 ][ 0 ] = 1;
11 for ( i = 1 ; i <= 181 ; ++ i )
12 for ( j = 1 ; j <= 4 ; ++ j )
13 for ( k = i*i ; k <= 32768 ; ++ k )
14 F[ j ][ k ] += F[ j-1 ][ k-i*i ];
15
16 int n;
17 while ( scanf("%d",&n) && n ) {
18 int sum = 0;
19 for ( i = 1 ; i <= 4 ; ++ i )
20 sum += F[ i ][ n ];
21 printf("%d\n",sum);
22 }
23 return 0;
24 }

posted on 2011-08-19 23:02  _xiaobai_  阅读(259)  评论(0编辑  收藏  举报