二分+貪心,這種題目一直都是自己的軟肋。
此題就是一個最大值最小化的問題,諸如此類的什么最大最小,最小最大之類的首先想到的應該是二分。
二分值,然后用貪心進行劃分,在劃分的時候需要注意分成k份,同時滿足前面的盡可能的少。思路簡單,實現起來有點麻煩。
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include <iostream>
using namespace std;
#define N 505
long long a[N];
int mk[N];
inline long long MAX(long long a, long long b)
{
return a > b ? a : b;
}
long long Judge(long long mid, const int m, const int k)
{
long long sum = 0, maxt = 0;
for(int i = m - 1, j = 0; i >= 0; i--)
{
if(j < k - 1 && (i < k - j - 1 || sum + a[i] > mid))
{
maxt = MAX(sum, maxt);
sum = a[i];
mk[i] = 1;
j++;
}
else sum += a[i];
}
return MAX(maxt, sum);
}
int main()
{
int t, m, k;
long long left, right, mid, ans;
scanf("%d", &t);
while(t--)
{
scanf("%d %d", &m, &k);
left = right = 0;
for(int i = 0; i < m; i++)
{
scanf("%lld", &a[i]);
right += a[i];
}
while(left < right)
{
mid = (left + right) >> 1;
ans = Judge(mid, m, k);
if(ans > mid) left = mid + 1;
else right = mid;
}
memset(mk, 0, sizeof(mk));
Judge(left, m, k);
printf("%lld", a[0]);
if(mk[0]) printf(" /");
for(int i = 1; i < m; i++)
{
printf(" %lld", a[i]);
if(mk[i]) printf(" /");
}
printf("\n");
}
return 0;
}