算法训练——礼物

学习分享

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

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

• 【代码如下】

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;
	}
}