1.题目给出数据范围,而且出现了最大值最小或最小值最大这样的字眼,所以采用二分
2.题目求的是每段的和的最大值,所以左边界为所有数据的最大值,右边界为所有数的和,即
cin >> n >> m;
for(int i = 1; i <= n; i++)
{
cin >> a[i];
l = max(a[i], l);
r += a[i];
}
3.二分出的结果需要判断是否符合要求,如果不符合要求,那么需要在左侧寻找,如果符合要求,那么在右侧寻找是否有更优的解,即
while(l <= r)
{
int mid = (l + r) / 2;
if(judge(mid))
{
l = mid + 1;
}
else
r = mid - 1 ;
}
4.judge函数为判断结果是否符合要求,因为二分的结果为每段和的最大,所以如果加上当前数超过这个值,就重新分一个段,否则就加上当前数,最后判断分的段与题目要求的分段数的关系,即
bool judge(int t)
{
int sum = 0, temp = 0;
for(int i = 1; i <= n; i++)
{
if(a[i] + temp > t)
{
sum++;
temp = a[i];
}
else
temp += a[i];
}
return sum >= m;
}
完整代码
#include <bits/stdc++.h>
using namespace std;
int n, m, a[100005], l = -1, r, ans;
bool judge(int t)
{
int sum = 0, temp = 0;
for(int i = 1; i <= n; i++)
{
if(a[i] + temp > t)
{
sum++;
temp = a[i];
}
else
temp += a[i];
}
return sum >= m;
}
int main(void)
{
cin >> n >> m;
for(int i = 1; i <= n; i++)
{
cin >> a[i];
l = max(a[i], l);
r += a[i];
}
while(l <= r)
{
int mid = (l + r) / 2;
if(judge(mid))
{
l = mid + 1;
}
else
r = mid - 1 ;
}
cout << l;
return 0;
}