算法训练 礼物（c++）

资源限制

时间限制：1.0s 内存限制：256.0MB

问题描述

　　JiaoShou在爱琳大陆的旅行完毕，即将回家，为了纪念这次旅行，他决定带回一些礼物给好朋友。
　　在走出了怪物森林以后，JiaoShou看到了排成一排的N个石子。
　　这些石子很漂亮，JiaoShou决定以此为礼物。
　　但是这N个石子被施加了一种特殊的魔法。
　　如果要取走石子，必须按照以下的规则去取。
　　每次必须取连续的2*K个石子，并且满足前K个石子的重量和小于等于S，后K个石子的重量和小于等于S。
　　由于时间紧迫，Jiaoshou只能取一次。
　　现在JiaoShou找到了聪明的你，问他最多可以带走多少个石子。

输入格式

　　第一行两个整数N、S。
　　第二行N个整数，用空格隔开，表示每个石子的重量。

输出格式

　　第一行输出一个数表示JiaoShou最多能取走多少个石子。

样列输入

　　8 3
　　1 1 1 1 1 1 1 1

样列输出

　　6

样列解释

　　任意选择连续的6个1即可。

数据规模和约定

　　对于20%的数据：N<=1000
　　对于70%的数据：N<=100,000
　　对于100%的数据：N<=1000,000，S<=10^12，每个石子的重量小于等于10^9，且非负

代码：因数据量过大使用二分查找方法进行求解：

#include<bits/stdc++.h>
using namespace std;
long long N, S; 
bool check(vector<long long> &sum, int mid){
	for(int i = mid; i<=N-mid; i++){
		if(sum[i] - sum[i-mid] <= S && sum[i+mid] - sum[i] <= S) 
		return true;
	}
	return false;
}
int main(){
	cin>>N>>S;
	vector<long long> sum(N+1, 0), num(N+1,0);
	for(int i=1; i<=N; i++){
		cin>>num[i];
		sum[i]=sum[i-1] + num[i];//前n项和 
	}
	int l=1, r=N;
	while(l<r){ //终止条件是l=r 
		int mid = (r+l+1)>>1;
		if(check(sum, mid)){
			l = mid; 
		}else{
			r = mid-1;
		}
	}
	cout<<2*l; 
}