POJ - 3104 Drying （二分搜索）

http://poj.org/problem?id=3104

题意：

有N件湿衣服，每件衣服的水分为ai个单位，每分钟自然减少1个单位。现有一烘干机，每分钟能烘一件衣服，每分钟下降K个单位的水分。水分为0时表示已被烘干。

问：最少要多少分钟才能将所有的衣服烘干。

数据范围比较大： (1 ≤ n ≤ 100 000). (1 ≤ a_i ≤ 10⁹),  (1 ≤ k ≤ 10⁹).

——————————————————————————

首先想到用优先队列模拟整个过程，但考虑到数据比较大，应该会超时。

如果用二分搜索的话，时间复杂度是O（nlog（MAX_a）），可以在2秒内解决的。

函数C（）为判断部分：

先将所有的衣服减去t个单位；

再将所有a[i] > 0 （还没烘干的）用烘干机烘干t--，如果时间t<0的话，表示时间太短了，return false。

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
const int MAX_N = 100000, INF = 1e9;
int N, K;
int a[MAX_N+10], aa[MAX_N+10]; 
bool C(int t)
{
	memcpy(aa, a, N*sizeof(a[0]));
	for(int i = 0; i < N; i++)
	{
		if(aa[i] > t)
			aa[i] -= t;
		else
			aa[i] = 0;
	}
	for(int i = 0; i < N; i++)
	{
		t -= aa[i] / (K-1);
		if(aa[i] % (K-1))
		{
			t--;
			aa[i] = 0;
		}
		if(t < 0)
			return false;
	}
	return true;
}
void solve()
{
	int Max = a[0];
	if(K == 1)
	{
		for(int i = 0; i < N; i++)
			Max = max(Max, a[i]);
		printf("%d\n", Max);
		return ;
	}
	int l = 0, r = INF;
	while(r - l > 1)
	{
		int mid = (l+r) / 2;
		if(C(mid))
			r = mid;
		else
			l = mid;
	}
	printf("%d\n", r);
}
int main()
{
	//freopen("in.txt", "r", stdin); 
	scanf("%d", &N);
	for(int i = 0; i < N; i++)
		scanf("%d", a+i);
	scanf("%d", &K);
	solve();

	return 0;
}