UVA 714-抄书(二分法)
题意:把一个包含n个正整数(不超过10^7)的序列划分成k个(1<=k<=n<=500)非空的连续子序列,每个正整数恰好是属于一个序列。S(i)表示第i个序列各个数之和,求所有S(i)的最大值尽量小的一种划分。如果有多个解,S(1)尽量小,如果仍有多解,S(2)尽量小,依此类推。
Sample Input
2
9 3
100 200 300 400 500 600 700 800 900
5 4
100 100 100 100 100
Sample Output
100 200 300 400 500 / 600 700 / 800 900
100 / 100 / 100 / 100 100
思路:
枚举一个数X,如果能把序列划分成k个或少于k个连续的子序列,使得所有的S(i)都不超过X,则答案小于或等于X;如果不能,则答案大于X。可以用二分加快枚举。
#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
typedef long long ll;
const int size = 500 + 5;
int n, k, a[size];
bool is[size];
int is_p(ll num)
{
ll tot = 0;
int cur_k = 1;
for(int i = 0; i < n; i++){
if(tot + a[i] <= num) tot += a[i];
else{
tot = a[i];
cur_k++;
}
}
return cur_k;
}
void print(ll num)
{
int rk = k - 1;
ll tot = 0;
memset(is, 0, sizeof(is));
for(int i = n - 1; i >= 0; i--){
if(tot + a[i] > num || i < rk){
tot = a[i];
is[i] = 1;
rk--;
}
else tot += a[i];
}
for(int i = 0; i < n-1; i++){
printf("%d ", a[i]);
if(is[i]) printf("/ ");
}
printf("%d\n", a[n-1]);
}
int main()
{
int T;
scanf("%d", &T);
while(T--){
scanf("%d%d", &n, &k);
int maxa = 0;
ll sum = 0;
for(int i = 0; i < n; i++){
scanf("%d", &a[i]);
maxa = max(maxa, a[i]);
sum += a[i];
}
ll l = (ll)maxa, r = sum;
while(l < r){
ll m = (l + r) / 2;
if(is_p(m) <= k) r = m;
else l = m + 1;
}
print(l);
}
return 0;
}
总结:
用二分法把优化问题转化为判定问题