1436:数列分段II -整型二分

1436:数列分段II

题目来源:一本通


【输入样例】

cpp 复制代码
5 3
4 2 4 5 1

【输出样例】

cpp 复制代码
6

题意

将数列分成若干段,最多M段,求这些段中最大值中的最小值。(M<=N是M的约束)

思路

  • 最大最小问题考虑二分。由于M越大,则每段的平均值越小,即里面的最大值的越小。M作为约束,所以当分段>M作为check()函数跳出条件
  • 题目没给数据范围,所以二分区间可以选择[min(a[i]),max(sum(a[i])](当然,简单点就直接设[0,1e10],不会超时),而M作为约束条件

数据约束

注意事项

哈哈,总有一个样例过不了。检查一下代码,发现有漏洞,始终要清醒:t必须是当前子段和最大值,但当a[i]>t,sum=a[i] 然后进入下一个循环,所以就有Bug。因此需要单独判断a[i],如果>t 就不符合条件。

eg:4 1 4 5 2 很显然有数据5,当Mid=4就不可能成立

参考代码

cpp 复制代码
#include<bits/stdc++.h>
using namespace std;
const int N =1000000; 
int n,m,a[N];
bool check(int t);//判断是否有m段 
int main()
{
	cin>>n>>m;
	int l = 0, r = 0;
	for(int i=0;i<n;i++){
		cin>>a[i];
		l = min(l,a[i]) ;
		r += a[i];
	}
	//二分
	int anwser = 0;
	while(l<=r){
		int mid = (r+l)/2;
		if(check(mid)) {
			anwser = mid;
			r= mid-1; //必须要区间保证是收敛的(不断缩小的) 
		}
		else l = mid+1;
	} 
	cout<<anwser;
   return 0;
}
bool check(int t){
	int sum = 0,ans = 1;//默认为a[0]容易有Bug 万一a[0]>m那就凉凉 
	for(int i=0;i<n;i++){
		if(a[i]>t){ //段里面的单个数据不能超过t否则出错 
			return false;
		}
		sum += a[i];
//		if(t ==5 ) cout<<sum<<"/"<<i<<"   "<<ans<<endl;
		//判断ans的情况,因为ans比m大,说明t可以分更小 ,也就是字段和不是最小值 
		if(sum>t){
			//因为t是sum里面的最大值,所以必须保证sum<=t 
			sum = a[i];
			ans++;
		}
		if(ans>m){
		   return false;
		}
	}
	return true;
}
相关推荐
拼命鼠鼠3 分钟前
【算法】矩阵链乘法的动态规划算法
算法·矩阵·动态规划
LYFlied19 分钟前
【每日算法】LeetCode 17. 电话号码的字母组合
前端·算法·leetcode·面试·职场和发展
程序喵大人33 分钟前
推荐个 C++ 练习平台
开发语言·c++·工具推荐
式5161 小时前
线性代数(八)非齐次方程组的解的结构
线性代数·算法·机器学习
fpcc1 小时前
跟我学C++中级篇——std::is_invocable的分析应
c++
橘颂TA2 小时前
【剑斩OFFER】算法的暴力美学——翻转对
算法·排序算法·结构与算法
叠叠乐2 小时前
robot_state_publisher 参数
java·前端·算法
hweiyu002 小时前
排序算法:冒泡排序
算法·排序算法
brave and determined2 小时前
CANN训练营 学习(day9)昇腾AscendC算子开发实战:从零到性能冠军
人工智能·算法·机器学习·ai·开发环境·算子开发·昇腾ai