首页 > 代码库 > shu_1299 vijos 1037(搭建双塔)

shu_1299 vijos 1037(搭建双塔)

http://202.121.199.212/JudgeOnline/problem.php?cid=1078&pid=7  


分析: 经典DP题

             dp[ i ][ j ]:前i个水晶搭建的高度差为 j 的双塔中较高塔的高度;

            

            dp[ i ][ j ] = d[ i-1 ][ j ]  ,                       不放第i个水晶;

            dp[ i ][ j ] = f[ i-1 ][ h[i]-j ] + j,               第i个水晶放在较矮的塔上,原来较矮的塔变成较高的塔;

            dp[ i ][ j ] = f[ i-1 ][ h[i]+j ],                    第i个水晶放在较矮的塔上,原来较矮的塔还是较矮的塔;

            dp[ i ][ j ] = f[ i-1 ][ j-h[i] ] + h[ i ],          第i个水晶放在较高的塔上,原来较高的塔变成更高的塔;

(可以画个图,转移方程就出来了)


代码:

#include <iostream>
#include <stdio.h>
#include <string>
#include <string.h>
using namespace std;
 
int h[120];
int dp[120][2400];
 
int mmax(int a,int b)
{
    return a<b ? b:a;
}
 
int main()
{
    //freopen("in.txt","r",stdin);
 
    int n;
    int sum=0;
    scanf("%d",&n);
    memset(dp, -1, sizeof(dp));
    dp[0][0]=0;
 
    for(int i=1;i<=n;i++) {scanf("%d",&h[i]); sum +=h[i];}
 
    for(int i = 1; i <=n; i++) {
        for(int j = 0; j <= sum; j++) {
            if(dp[i-1][j]>-1)
               dp[i][j]=dp[i-1][j];
            if(h[i]>j && dp[i-1][h[i]-j]>-1)
               dp[i][j]=mmax(dp[i][j],dp[i-1][h[i]-j]+j);
            if(j+h[i]<=sum && dp[i-1][j+h[i]]>-1)
               dp[i][j]=mmax(dp[i][j],dp[i-1][j+h[i]]);
            if(j>=h[i] && dp[i-1][j-h[i]]>-1)
               dp[i][j]=mmax(dp[i][j],dp[i-1][j-h[i]]+h[i]);
        }
    }
    if(dp[n][0] > 0)   printf("%d\n",dp[n][0]);
    else               printf("Impossible\n");
    return 0;
}