题目描述
OJ地址 http://lx.lanqiao.cn/problem.page?gpid=T449
小明需要在一篇文档中加入 N 张图片,其中第 i 张图片的宽度是 Wi,高度是 Hi。
假设纸张的宽度是 M,小明使用的文档编辑工具会用以下方式对图片进行自动排版:
1. 该工具会按照图片顺序,在宽度 M 以内,将尽可能多的图片排在一行。该行的高度是行内最高的图片的高度。例如在 M=10 的纸张上依次打印 3x4, 2x2, 3x3 三张图片,则效果如下图所示,这一行高度为4。(分割线以上为列标尺,分割线以下为排版区域;数字组成的矩形为第x张图片占用的版面)
2. 如果当前行剩余宽度大于0,并且小于下一张图片,则下一张图片会按比例缩放到宽度为当前行剩余宽度(高度向上取整),然后放入当前行。例如再放入一张4x9的图片,由于剩余宽度是2,这张图片会被压缩到2x5,再被放入第一行的末尾。此时该行高度为5:
3. 如果当前行剩余宽度为0,该工具会从下一行开始继续对剩余的图片进行排版,直到所有图片都处理完毕。此时所有行的总高度和就是这 N 张图片的排版高度。例如再放入11x1, 5x5, 3x4 的图片后,效果如下图所示,总高度为11:
现在由于排版高度过高,图片的先后顺序也不能改变,小明只好从 N 张图片中选择一张删除掉以降低总高度。他希望剩余N-1张图片按原顺序的排版高度最低,你能求出最低高度是多少么?
输入:
第一行包含两个整数 M 和 N,分别表示纸张宽度和图片的数量。
接下来 N 行,每行2个整数Wi, Hi,表示第 i 个图大小为 Wi*Hi。
对于30%的数据,满足1<=N<=1000
对于100%的数据,满足1<=N<=100000,1<=M, Wi, Hi<=100
输出:
一个整数,表示在删除掉某一张图片之后,排版高度最少能是多少。
样例输入1:
4 3
2 2
2 3
2 2
样例输出1:
2
样例输入2:
2 10
4 4
4 3
1 3
4 5
2 1
2 3
5 4
5 3
1 5
2 4
样例输出2:
17
思路:
我用dfs实现的,总的思路就是,把所有的图片循环一遍,依次抽出来不参与排版,剩下的n-1张图片进行排版。
dfs时的参数有,要给第几张照片排(st),当前行已经排了多宽(w),当前行的高度(h),还有当前行上边所有行的高度和(sum_h)(因为只要当前行没排满,那么这一行的高度暂时就无法确定,所以用之前所有行的高度和),一共这四个参数。
要排当前第st张图片的时候,考虑:如果不进行缩放的话,当前行能否放下它。如果放得下,直接放,然后dfs安排下一个图片。如果放不下,就对这张图片进行缩放处理,然后再dfs下一张图片。
这张图排好以后,如果当前行满了,则下一次dfs时的参数st应该+1,w 和 h应该为0,sum_h应该为sum_h加上最后一张图片的高度(注意是否缩放)。
正常的dfs做到如上步骤就可以了,但是这个题还可以进行剪枝优化。
把第 i 张图片抽出来不参与排版的话,前 i - 1 张图片的排版是没有变化的,只需要对 i + 1 以及之后的图片进行重新排版。这需要用一个数组 data记录每个图片排完以后的状态(当前图片排完以后,当前行宽度、高度、之前的总高度)。所以在抽出第 i 张图片不参与排版时,接着 第 i -1 张的排完后的状态开始排第 i + 1张以及以后的图片。这一块,仔细看代码中的注释。
另外一个方面的剪枝,利用记忆化搜索,如果第 i 张图片以前排到过某行的第一个位置,那么他后边的排法肯定不变了,后边的高度和也不会变。所以这就需要先把所有的图片(n张)先排一遍,用数组 t 记录一下状态,t[i] = x 表示 第 i 张图片排在了某一行的第一张时,他当前所在的行以及他后边所有行的高度和。在dfs时,如果遇到某张图要排在某行的第一个位置了,就可以直接返回后边的高度了。
代码
#include <bits/stdc++.h>
using namespace std;
int m,n,ans = 999999999;
bool f = true;
int t[100005];
struct nn{
int w,h;
} pic[100005];
struct state{
int w,h,sum_h;
} data[100005];
void dfs(int st, int w,int h, int sum_h){ //从哪开始取图片 已经排的宽度、当前行最高高度 本行以前的总高度
if(w == 0 && t[st] != -1){
ans = min(ans,sum_h + t[st]);
return ;
}
if(st == n+1){ // 第n张排完了
if(w == 0){ //第n张排在了某行的最后 sum_h已经计入了这一张的高度,sum_h已经是总的高度了
ans = min(ans,sum_h);
}else{ // 第n张图片没把这行排满,所以sum_h 没记录当前行的高度,所以是sum_h+h才对
ans = min(ans,sum_h+h);
}
return ;
}
// 开始排版
if(w + pic[st].w > m){
//不能直接放下最后一个 需要缩放
int d = m - w; //还剩下d这么宽
int cur_h = 0; // 最后一张的高度 把这行填满
if((pic[st].h * d) % pic[st].w == 0){
cur_h = (pic[st].h * d) / pic[st].w;
}else{
cur_h = ((pic[st].h * d) / pic[st].w) + 1;
}
if(f){
data[st].w = m;
data[st].h = max(h,cur_h);
data[st].sum_h = sum_h+max(h,cur_h);
}
dfs(st+1,0,0,sum_h+max(h,cur_h));
}else if( w + pic[st].w == m ){
//正好放下这张
if(f){
data[st].w = m;
data[st].h = max(h,pic[st].h);
data[st].sum_h = sum_h + max(h,pic[st].h);
}
dfs(st+1,0,0,sum_h+max(h,pic[st].h));
}else{
//cout << "当前行可以放下 " << st << " 且有剩余空间"<< endl;
if(f){
data[st].w = w+pic[st].w;
data[st].h = max(h,pic[st].h);
data[st].sum_h = sum_h + max(h,pic[st].h);
}
dfs(st+1,w+pic[st].w,max(h,pic[st].h),sum_h );
}
}
int main(){
/*
* 去掉哪一张后,其他的n-1张图片排版的高度最低,求出这个最低高度。
*/
memset(t,-1,sizeof(t));
cin >> m >> n;
for(int i = 1 ; i <= n; i++){
scanf("%d%d",&pic[i].w,&pic[i].h);
}
// 先排一遍,然后往出抽取,继续前一个照片的位置开始dfs
dfs(0,0,0,0);
f = false;
t[1] = data[n].sum_h;
for(int i = 2; i <= n ;i++){
if(data[i-1].w == m){ // 第i个放完就满行了
t[i] = t[1] - data[i-1].sum_h; //第i+1张 放在行首 后边的高度
}
}
for(int i = 1; i <= n; i++){
// 抽取 接着i-1的开始排列
if(i == 1){
dfs(2,0,0,0);
}else{
if(data[i-1].w == m){
// 换行
dfs(i+1,0,0,data[i-1].sum_h);
}else{
/*
第 i 张图片不能直接接着i - 1的结果排,因为 dfs的第四个参数是不包括当前行的高度总和,
而data中存的那个sum_h是包括当前行和以前的总高度,所以
需要找到 第 i 张图片以前,比如 第 j 张图片把上一行排满了,传入dfs的第四个参数应该是上data[j].sum_h。
这里需要多想一下,或者说,第 i 张图片排的话,可能不是排在一行的第一个位置而是排在第四个位置,那么
前3个位置排完后的高度,我们无法知道。
*/
// tmp是前一行填满后的高度 从i往前倒着找 看看哪一张图片排完后,那一行排满了
int tmp = 0;
for(int j = i-1; j >=1; j--){
if(data[j].w == m){
tmp = data[j].sum_h;
break;
}
}
dfs(i+1,data[i-1].w,data[i-1].h,tmp);
}
}
}
cout << ans << endl;
return 0;
}