【SSL 1554】抄书(二分答案) (区间DP)

题目大意

    某寺院要求僧人抄写经书，现在把 n本经书放成一排，分给m个僧人抄写，每个人的抄写速度是一样的，一本书只能给一个人抄写，而且每人抄写的书必须是连续的（比如不能把第1本、第3本给同一个人抄写），最后完成抄写的人所花费的时间就是抄写时间。

     现在告诉你每本书的页数，请你设计一种方案，使得抄写的时间最少。

输入格式

第一行两个整数 n，m 。（n≤m≤500）
第二行n个整数，表示每本书的页数。

输出格式

共 m行，每行两个整数，第i行表示第i个人抄写的书的起始编号和终止编号。
m行的起始编号应该从小到大排列，如果有多解，则尽可能让前面的人少抄写。

输入样例

9 3
1 2 3 4 5 6 7 8 9

输出样例

1 5
6 7
8 9

基本思路

这题看似难搞，实际上摸清楚了特别好做。

大体思路是先求解出答案，再用贪心进行分配。在后面的僧侣则尽量多分配，对后面的僧侣一直分配直至超出答案。状态也十分好分析，用DP的经典思想：看前 i个考虑当前。f[i][j]意思便是i个人抄完前j本书所用时间。

大家有没有发现这题的问法是典型的"使最大最小"，抄书时间是用时最长的那个僧侣所用时间嘛，于是这题实际上是可以用二分做的。因为最终答案是单独处理的其他部分便无需过多改动。

核心代码

区间DP

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=510;
int n,m,a[N],s[N],f[N][N];
//f[i][j]为前i个人抄完前j本书所用时间
//先算出时间,再用贪心寻找方案 
void print(int p,int w){
	//从最后一个人找起,只要不超
	//过答案就一直抄下去
	if(p<=0) return;
	int sum=0,i=w;
	while(i>=1){
		sum+=a[i];
		if(sum>f[m][n]) break;
		i--;
	}
	print(p-1,i);
	cout<<i+1<<" "<<w<<endl;
}
int main(){
	ios::sync_with_stdio(false);
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		s[i]=a[i]+s[i-1];
	}
	memset(f,0x3f,sizeof(f));
	//DP
	for(int i=1;i<=n;i++) f[1][i]=s[i];
	for(int p=2;p<=m;p++){//人
		for(int i=p;i<=n;i++){//抄到第i本书
			for(int j=p;j<=i;j++){//从第j本书开始抄
				//j表示抄多少本也可以,但如此一来方便计算前缀和
				f[p][i]=min(f[p][i],max(f[p-1][j-1],(s[i]-s[j-1])));
			}
		}
	}
	print(m,n);
	return 0;
}

二分答案

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=510;
int n,m,a[N],l,r,mid;
//f[i][j]为前i个人抄完前j本书所用时间
//先算出时间,再用贪心寻找方案 
void print(int p,int w){
	//从最后一个人找起,只要不超
	//过答案就一直抄下去
	if(p<=0) return;
	int sum=0,i=w;
	while(i>=1){
		sum+=a[i];
		if(sum>l) break;
		i--;
	}
	print(p-1,i);
	cout<<i+1<<" "<<w<<endl;
}
bool check(int x){
	//x为所用时间 
	int sum=1,t=a[1];
	for(int i=2;i<=n;i++){
		t+=a[i];
		if(t>x){
			sum++;
			t=a[i];
		}
	}
	if(sum>m) return false;
	return true;
}
int main(){
	ios::sync_with_stdio(false);
	cin>>n>>m;l=r=0;
	for(int i=1;i<=n;i++){
		cin>>a[i];
		l=max(l,a[i]);
		r+=a[i];
	}
	//二分答案 
	while(l<r){
		mid=(l+r)>>1;
		if(check(mid)==true) r=mid;
		else l=mid+1;//太小了 
	}
	print(m,n);
	return 0;
}