蓝桥杯算法训练—— 礼物 Java

问题描述

　　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，且非负

由于N值太大，如果运用Scanner来读取数据的话，在提交会面临超时及内存超限的情况，用BufferedReader的话可以很好的规避这个问题。

这道题目我用的是二分法，暴力的话大概只能拿前面百分之二十的分。

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main{
	static int count =0;
	static long S;
	public static void main(String args[]) {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		String s,s2;
		String[] b1 = null,b2 = null;
		//BufferedReader按行读取，对读取数据进行切片并存储
		try {
				s=br.readLine();
				b1=s.split(" ");
				s2=br.readLine();
				b2=s2.split(" ");
	
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
		int N=Integer.parseInt(b1[0]);
		S=Long.parseLong(b1[1]);
		
		int[] a=new int[N+1];
		long[] sum=new long[N+1];
		for(int i=1;i<=b2.length;i++) {
			a[i]=Integer.parseInt(b2[i-1]);
			//sum[n],代表前n个石子重量
			sum[i]=sum[i-1]+a[i];
		}
		int l=1,r=N;
		//mid代表题目中的k个石子
		while(l<r) {
			//二分查找寻找最适合的k取值
			int mid=(l+r+1)/2;
			if(fun(mid,N,sum)) {
				//代表k的取值即mid过大，可尝试扩大范围
				l=mid;
			}else {
				//代表k的取值即mid过大，缩小范围
				r=mid-1;
			}
		}
		//最佳的k值为l，总共的石子为2*l
		System.out.print(2*l);
	}
	public static boolean fun(int mid,int n,long[] sum) {
		//对sum数组查询，若存在i前mid个石子与后mid个石子的重量皆小于S,则返回true,代表可进一步扩大mid即k的范围
		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;
	}
}