【DFS】蓝桥杯试题 图形排版

题目描述

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;
}