题目:给你一些砖块,问你是否能罗列成2个高度相同的塔,每层一个石头。
分析:dp,双塔问题。和LIS,背包等问题相同,前 i项的最优子问题。
状态:f(i,j)为前 i个材料,在两塔差的绝对值为j时的高塔(或者低塔)的高度;
决策:每次有 3种选择:放在高塔上,放在低塔上,或者不放;
T = O( sum( h )*n ) { 阶段数*长度总和/2 }。
#include <stdio.h> #include <stdlib.h> #include <string.h> #include <math.h> #define max( a, b ) ((a)>(b)?(a):(b)) int f[ 1001 ]; int t[ 1001 ]; int h[ 101 ]; int main() { int n,s; while ( scanf("%d",&n) != EOF && n != -1 ) { s = 0; for ( int i = 1 ; i <= n ; ++ i ) { scanf("%d",&h[ i ]); s += h[ i ]; } s /= 2; memset( f, 0, sizeof( f ) ); memset( t, 0, sizeof( t ) ); for ( int i = 1 ; i <= n ; ++ i ) { for ( int j = 0 ; j <= s ; ++ j ) if ( t[ j ] ) { f[ j+h[ i ] ] = max( t[ j ]+h[ i ], f[ j+h[ i ] ] ); f[ abs(h[ i ]-j) ] = max( max( t[ j ], t[ j ]+h[ i ]-j ), f[ abs(h[ i ]-j) ] ); } f[ h[ i ] ] = max( h[ i ], f[ h[ i ] ] ); for ( int j = 0 ; j <= s ; ++ j ) f[ j ] = max( f[ j ], t[ j ] ); for ( int j = 0 ; j <= s ; ++ j ) t[ j ] = f[ j ]; } if ( f[ 0 ] ) printf("%d\n",f[ 0 ]); else printf("Sorry\n"); } return 0; }