Time Limit:1000MS Memory Limit:32768KB 64bit IO Format:%lld & %llu
Description
This winter we are going on a trip to Bandorban. The main target is to climb up to the top of Keokradong. So, we will use a trail. The trail is a continuous marked footpath that goes from Bandorban to Keokradong.
Part of the experience is also the route planning of the trip. We have a list of all possible campsites that we can use along the way and we want to do this trip so that we only stop K nights to camp. We also know in advance the distance between consecutive campsites and we are only allowed to camp at a campsite. Our goal is to plan the trip so that we minimize the maximum amount of walking done in a single day. In other words, if our trip involves 2 nights (3 days of walking), and we walk 9, 10, 5 miles on each day respectively, the cost (maximum amount of walking done in one day) is 10. Another schedule that involves walking 9, 6, 9 miles on each day has cost 9.
Given the distances between N consecutive campsites of a trail and given the number of nights for your trip, K, your task is to devise a camping strategy for the specified trail such that it minimizes the maximum amount of walking done in a single day. Note that the first distance value given is the distance from our start-point of the trail to our 1st campsite, and the last distance value given is the distance from our Nth campsite to our end-point of the trail.
Input
Input starts with an integer T (≤ 200), denoting the number of test cases.
Each case contains of two integers, the number of campsites, N (1 ≤ N ≤ 1000) and the number of nights of the trip, K (1 ≤ K ≤ min(N, 300)). The following N + 1 lines indicate the distance in miles between consecutive campsite locations. All the integers will be positive and less than10000.
Output
For each case of input you have to print the case number and the minimized cost as described above. Then print K+1 lines, each containing the amount of distance covered in ith day. As there can be many solutions, the primary target is to find the one which ensures that each day we have to walk some distance. For ties, print the one where the distance covered in first day is maximum, then the distance covered in second day is maximum and so on.
Sample Input
1
4 3
7
2
6
4
5
Sample Output
Case 1: 8
7
8
4
5
#include<stdio.h>
#include<string.h>
#include<algorithm>
const int inf = 0x3f3f3f3f ;
int n , m ;
int T ;
int a[] ;
int l , r ; bool solve (int mid)
{
int tmp = , cnt = ;
for (int i = ; i < n ; i ++) {
tmp += a[i] ;
if (tmp > mid) {
tmp = a[i] ;
cnt ++ ;
}
}
cnt ++ ;
return cnt > m ;
} int main ()
{
//freopen ("a.txt" , "r" , stdin ) ;
scanf ("%d" , &T) ;
int cas = ;
while (T --) {
scanf ("%d%d" , &n , &m ) ;
n ++ ; m ++ ;
l = - inf ;
r = ;
for (int i = ; i < n ; i ++) {
scanf ("%d" , &a[i]) ;
l = std::max (a[i] , l ) ;
r += a[i] ;
}
while (l <= r) {
int mid = ( l + r ) / ;
if (solve (mid) ) l = mid + ;
else r = mid - ;
}
printf ("Case %d: %d\n" , cas ++ , l ) ;
int sum = ;
int cnt = ;
for (int i = ; i < n ; i ++) {
sum += a[i] ;
if (sum > l) {
printf ("%d\n" , sum - a[i]) ;
sum = a[i] ;
cnt ++ ;
}
if (m - cnt + i >= n) {
printf ("%d\n" , sum ) ;
for (int j = i + ; j < n ; j ++) printf ("%d\n" , a[j]) ;
break ;
}
}
}
return ;
}
initial l = max{a[]} , r = sum {a[]} ;
很明显我们要求的x肯定在[l,r]这个区间内。
我们也能很容易求出：当组合后各个堆中最大的x已知时，至少需要走的天数 day。
所以我们令mid = (l + r）/ 2 ; 并求出对应的 day，if day > (k + 1) , 说明x的值应在[mid + 1,r]上 ； else ， 便在[l,mid - 1]上（ps：至于为什么day == k + 1是也定位在这一块，是因为我们想令day “minimize”）。
而且因为我们所求的最后x，是指至少需要走的天数，所以很多情况下会比k + 1小，所以输出时应尽量让前面的a[i]相加贴近x，在最后到。。。就单个输出啊a[i]来补到k+1个。