这个题巧妙运用了二分然后逼近。首先确定min(S)的范围,然后再用二分逼近,最后的结果就是必然存在的那个最小值。
另外,输出使用了贪心。而且这里的板是“后置板”,即先输出数,在输出板。位置i最多可以分成i+1块,而此刻的需求k是至少让剩余部分分成k-1块(因为如果在这里加一块的话,后面的部分会加一块)。如果i+1==k-1,即i+1<k的时候,就需要让后面的部分每一个单独的元素变成一块了。
#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
ll a[505];
bool ban[505];
int n;
ll solve(ll q){
ll he=0,ans=1;
for(int i=0;i<n;i++){
if(he+a[i]>q){
he=a[i];
ans++;
}
else{
he+=a[i];
}
}
return ans;
}
void print(ll q,ll k){
ll he=0;
memset(ban,false,sizeof(ban));
for(int i=n-1;i>-1;i--){
if(he+a[i]>q||i<k-1){
ban[i]=true;
he=a[i];
k--;
}
else{
he+=a[i];
}
}
for(int i=0;i<n-1;i++){
// printf(i==n-1?"%lld":"%lld ",a[i]);
printf("%d ",a[i]);
if(ban[i]){
// printf(i==n-1?"\n":"/ ");
printf("/ ");
}
}
printf("%d\n",a[n-1]);
}
int main(void){
int t;
cin>>t;
while(t--){
ll k;
cin>>n>>k;
ll mxa=0,tot=0;
for(int i=0;i<n;i++){
scanf("%lld",&a[i]);
tot+=a[i];
mxa=max(mxa,a[i]);
}
ll q,L=mxa,R=tot;
while(L<R){
q=L+(R-L)/2;
if(solve(q)>k){
L=q+1;
}
else{
R=q;
}
// printf("L=%lld,R=%lld\n",L,R);
}
print(L,k);
}
return 0;
}