shopee的零食柜

题目描述

  shopee的零食柜，有着各式各样的零食，但是因为贪吃，小虾同学体重日益增加，终于被人叫为小胖了，他终于下定决心减肥了，他决定每天晚上去操场跑两圈，但是跑步太累人了，他想转移注意力，忘记痛苦，正在听着音乐的他，突然有个想法，他想跟着音乐的节奏来跑步，音乐有7种音符，对应的是1到7，那么他对应的步长就可以是1-7分米，这样的话他就可以转移注意力了，但是他想保持自己跑步的速度，在规定时间m分钟跑完。为了避免被累死，他需要规划他每分钟需要跑过的音符，这些音符的步长总和要尽量小。下面是小虾同学听的歌曲的音符，以及规定的时间，你能告诉他每分钟他应该跑多少步长？

输入描述

  输入的第一行输入 n（1 ≤ n ≤ 1000000，表示音符数），m（1<=m< 1000000, m <= n）组成，第二行有 n 个数，表示每个音符（1<= f <= 7)。

输出描述

输出每分钟应该跑的步长。

解题思路

  这道题其实是相当于在N个数间插入M-1个棍，将其分为M组，分组的和的最大值就是这一种分法的步长，求所有分法中步长最小的那一种。
  暴力搜索的方法想当于在N-1个位置上挑出M-1个位置进行全排列，这样时间复杂度会很高，尝试过深度搜索之后果然超时。
  换一个角度;将这N个数分为M组，所有组的和是数据的总和，分组越多，每一组的大小越小，步长越小，分组越少，每一组的大小越大，步长越大，最小的步长是所有数据中的最大值（分组为N），最大的步长是所有数的总和（分组为1），那么题目就是寻找满足分成M组的最小步长，有了上下界，就可以进行二分搜索。

实现代码

#include <iostream>
#include <vector>
using namespace std;
class Solution{
	public:
	int getMin(vector<int>& data,int n,int m){
		int sum=0;
		for(int i=0;i<n;i++)
			sum+=data[i];	
		int low=sum/m,high=sum,mid;
		while(low<high){
			mid=(low+high)/2;
			int cnt=0,i=0,tmp;
			while(i<n){
				tmp=0;
				while(i<n&&tmp+data[i]<=mid){
					tmp+=data[i];
					i++;
				}
				cnt++;
			}
			if(cnt<=m) high=mid;
			else  low=mid+1;		
		}
		return low;
	}
};
int main(){
	int n,m;
	cin>>n>>m;
	vector<int> data(n,0);
	for(int i=0;i<n;i++)
		cin>>data[i];
	Solution s;
	cout<<s.getMin(data,n,m)<<endl;
}

  二分搜索有两种可能的结果，一种是分为M组的最小步长，刚好就是解，还有一种是由于当遍历到这个结果之后，二分法还没有结束（low和high之间还有距离)，这个时候再向下进行，找到的可能就是大于M组的步长，所以我保持了一个分为M组的时候的记录minv，但是测试结果没有直接cout<<mid<<endl的好，我怀疑这是测试用例的问题，或者是我有所疏漏，欢迎大家指教。