洛谷P1182 数列分段 Section II
题目描述
对于给定的一个长度为 N 的正整数数列 A1∼N,现要将其分成M(M≤N)段,并要求每段连续,且每段和的最大值最小。
关于最大值最小:
例如一数列 4 2 4 5 1要分成 3 段。
将其如下分段:
[4 2][4 5][1]
第一段和为 6,第 2 段和为 9,第 3 段和为 1,和最大值为 9。
将其如下分段:
[4][2 4][5 1]
第一段和为 4,第 2 段和为 6,第 3 段和为 6,和最大值为 6。
并且无论如何分段,最大值不会小于 6。
所以可以得到要将数列 4 2 4 5 1 要分成 3 段,每段和的最大值最小为 6。
输入格式
第 1 行包含两个正整数 N,M。
第 2 行包含 N 个空格隔开的非负整数 Ai,含义如题目所述。
输出格式
一个正整数,即每段和最大值最小为多少。
输入输出样例
输入 #1
5 3 4 2 4 5 1
输出 #1
6
说明/提示
对于 20% 的数据,N≤10。
对于 40% 的数据,N≤1000N≤1000。
对于 100% 的数据,1≤N≤10^5,M≤N,Ai<10^8, 答案不超过 10^9。
思路:
好经典的二分,套个模板,完事!
具体思路:
1.定义数组和变量
2.定义一个isok函数,用于判断这个最大值是否可行
函数里面写:
1.定义cnt=0,sum=0,cnt用来存段数,sum用来存每一段的和
2.遍历数组
3.正常二分
AC代码:
#include <bits/stdc++.h>
using namespace std;
int a[100005];
int n,m;
bool isok(int x)
{
int cnt=0,sum=0;
for(int i=1;i<=n;i++)
{
if(a[i]>x)
{
return 0;
}
if(sum+a[i]<=x)
{
sum+=a[i];
}
else
{
sum=a[i];
cnt++;
}
}
return cnt+1<=m;
}
int main()
{
cin >> n >> m;
int l=0,r=0;
for(int i=1;i<=n;i++)
{
cin >> a[i];
r+=a[i];
}
while(l<=r)
{
int mid=l+r>>1;
if(isok(mid))
{
r=mid-1;
}
else
{
l=mid+1;
}
}
cout<<l;
return 0;
}