DAG上动态规划——巴比伦塔问题

有n种立方体,每种都有无穷多个,要求选一些立方体摞成一根尽量高的柱子,可以自行选一边当多高,使得每个立方体的底面长宽分别严格小于它下方的立方体的底面长宽。
题目链接

这个问题可以转化为一个DAG问题,与矩形嵌套问题类似。
转化过程:将一个给定的立方体分为三个分别以不同的边为高的立方体,假如给定立方体为 ( 1 , 2 , 3 ) (1,2,3) (1,2,3),分成以第一个数为高的三个立方体: ( 1 , 2 , 3 ) (1,2,3) (1,2,3) ( 2 , 1 , 3 ) (2,1,3) (2,1,3) ( 3 , 2 , 1 ) (3,2,1) (3,2,1)(注意: ( 2 , 3 , 1 ) (2,3,1) (2,3,1) ( 2 , 1 , 3 ) (2,1,3) (2,1,3)是相同的状态),这样任何一个节点就不可能指向自己。
假如给了30个立方体,就可以组成一个有90个节点的有向无环图,假如 X 状态可以放在 Y 状态的上面,则说 X 指向 Y 。然后可以给这 90 个立方体状态编号。

        for(int ii=0;ii<n;ii++){
            int i=ii*3;
            cin >> nn[0] >> nn[1] >> nn[2];
            for(int j=0;j<3;j++){//数组第一个数表示高,二三表示长和宽
                towerNode[i+j][0]=nn[j];
                towerNode[i+j][1]=max(nn[(j+1)%3],nn[(j+2)%3]);
                towerNode[i+j][2]=min(nn[(j+1)%3],nn[(j+2)%3]);
            }
        }

建立图:

        for(int i=0;i<n*3;i++){
            for(int j=0;j<n*3;j++){
                if(towerNode[i][1]<towerNode[j][1] && towerNode[i][2]<towerNode[j][2]){
                    G[i][j]=1;
                }
            }
        }

使用记忆化搜索的方法寻找最长路径。
状态转移方程为:

d ( i ) = m a x { d ( j ) + f ( i ) ∣ ( i , j ) ∈ E } d(i)=max\left \{ d(j)+f(i)|(i,j)\in E \right \} d(i)=max{d(j)+f(i)(i,j)E}
f ( i ) f(i) f(i) i i i 节点立方体的高。

int dp(int i){
    int& ans = d[i];
    if(ans > 0)
        return ans;
    ans = towerNode[i][0];
    for(int j=0;j<n*3;j++){
        if(G[i][j])
            ans = max(ans,dp(j)+towerNode[i][0]);
    }
    return ans;
}

完整程序:

#define LOCAL
#include <iostream>
#include <cstring>
#include <string>
#include <cstdio>
//#include <stdio.h>
#include <cstdlib>
#include<time.h>
#include <ctype.h>
#include <sstream>
#include <assert.h>
#include <math.h>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <stack>
#include <queue>
#define MAXN 92

using namespace std;

int n;
//保存图的邻接矩阵
int G[MAXN][MAXN];
int d[MAXN];

//以数组来记录图中的点,数组第一个数表示高,二三表示长和宽
int towerNode[MAXN][3];



//计算i节点的深度
int dp(int i){
    int& ans = d[i];
    if(ans > 0)
        return ans;
    ans = towerNode[i][0];
    for(int j=0;j<n*3;j++){
        if(G[i][j])
            ans = max(ans,dp(j)+towerNode[i][0]);
    }
    return ans;
}

int main()
{
    #ifdef LOCAL
    freopen("data.in","r",stdin);
    freopen("data.out","w",stdout);
    #endif // LOCAL
    int nn[3];
    int N=0;
    while(cin >> n && n){
        N++;
        memset(G,0,sizeof(G));
        memset(d,-1,sizeof(d));
        for(int ii=0;ii<n;ii++){
            int i=ii*3;
            cin >> nn[0] >> nn[1] >> nn[2];
            for(int j=0;j<3;j++){//数组第一个数表示高,二三表示长和宽
                towerNode[i+j][0]=nn[j];
                towerNode[i+j][1]=max(nn[(j+1)%3],nn[(j+2)%3]);
                towerNode[i+j][2]=min(nn[(j+1)%3],nn[(j+2)%3]);
            }
        }
        for(int i=0;i<n*3;i++){
            for(int j=0;j<n*3;j++){
                if(towerNode[i][1]<towerNode[j][1] && towerNode[i][2]<towerNode[j][2]){
                    G[i][j]=1;
                }
            }
        }
        for(int i=0;i<n*3;i++){
            dp(i);
        }
        cout << "Case " << N << ": maximum height = " << *max_element(d,d+MAXN) << "\n";
    }
  return 0;
}

  • 0
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值