作者:xq的acm之路
贪心+二分
题目链接:https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=655
题目大意:把一个包含m个正整数的序列划分成k个非空的连续子序列,使得每个正整数恰好属于一个序列。设第i个序列的各数之和为S(i),你的任务是让所有S(i)的最大值尽量小。
思路:
“最大值尽量小”问题。
区间的范围肯定是所有数中最大的一个至所有数之和,我们可以使用二分法来确定这样的一个值x,x尽量小并且每个区间都不大它。
题目要求的是前面的区间尽量小,那我们就需要贪心一下,从右边的数开始分配区间。需要注意的是,如果当前剩下来的待分配区间的数正好等于还要分配的组数,那么前面的几个数每个数都为一个区间。
题意懒得打,就摘了网上的。。。。
代码如下:
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int a[10000005],b[510];
int m,k;
long long sum,Max;
bool panduan(long long mid)
{
long long num=0;
int jishu=1;
for(int i=m-1; i>=0; i--)
{
if(num+a[i]>mid)
{
num=a[i];
jishu++;
if(jishu>k) return false;
}
else
num+=a[i];
}
return true;
}
void solve()
{
memset(b,0,sizeof(b));
long long left=Max,right=sum;
while(left<=right)
{
long long mid=(left+right)/2;
if(panduan(mid))
{
right=mid-1;
}
else
left=mid+1;
}
//cout<<left<<endl;
int jishu=1;
long long sum=0;
for(int i=m-1; i>=0; i--)
{
if(sum+a[i]>left)
{
sum=a[i];
jishu++;
b[i]=1;
}
else
sum+=a[i];
if(k-jishu==i+1)
{
for (int j = 0; j <= i; j++)
b[j] = 1;
break;
}
}
//for(int i=0;i<m-1;i++)
//cout<<b[i]<<" ";
for(int i=0;i<m-1;i++)
{
cout << a[i] << " ";
if (b[i]) cout << "/ ";
}
cout<<a[m-1]<<endl;
}
int main()
{
int t;
cin>>t;
while(t--)
{
cin>>m>>k;
Max=0,sum=0;
for(int i=0; i<m; i++)
{
cin>>a[i];
if(a[i]>Max)
Max=a[i];
sum+=a[i];
}
solve();
}
return 0;
}