蓝桥杯算法训练-礼物（二分）

资源限制

时间限制：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，且非负

这题用Java内存超限，用c++就可以AC......

下面是Java的代码，C++也是同理

import java.util.*;

/*
 * 思路：
 * 	这题我想不出来什么思路，参考网上的做法，先求前缀和，再用二分法找答案
 * 	具体做法：
 * 		首先求出石子数组的前缀和
 * 		再用二分法查找出对应长度的序列是否符合要求，查找出序列长度最大的。
 * 		
 * */
public class T1003 {
	
	static int N;
	static long S;
	static long[] stone;
	static long[] preFix;
	
	public static void main(String[] args) {
		Scanner scanner = new Scanner(System.in);
		N = scanner.nextInt();
		S = scanner.nextLong();
		stone = new long[N + 1];
		preFix = new long[N + 1];
		//求前缀和
		for(int i = 1; i <= N; i++) {
			stone[i] = scanner.nextLong();
			preFix[i] = preFix[i - 1] + stone[i];
		}
		int left = 1;
		int right = N;
		//用二分法求
		while(left < right) {
			int mid = left + right + 1 >> 1; //相当于除以二
			if(check(mid)) {
				left = mid;
			}else {
				right = mid - 1;
			}
		}
		System.out.println(2 * left);
	}

	private static boolean check(int mid) {
		for(int i = mid; i <= (N - mid); i++) {
			if(preFix[i] - preFix[i - mid] <= S && preFix[i + mid] - preFix[i] <= S) {
				return true;
			}
		}
		return false;
	}
}