问题描述:
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
数据规模和约定
对于20%的数据:N<=1000
对于70%的数据:N<=100,000
对于100%的数据:N<=1000,000,S<=10^12,
每个石子的重量小于等于10^9,且非负
思路:
由题意很容易想到前缀和。
前缀和:
就是数列的前n项和
接收数据:
对于有大量输入数据的情况可以考虑用BufferdReader来代替Scanner,这样会节省大量运行时间,我之前使用的Scanner,它报运行超时,改为BufferedReader后就100分了。
BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
String[] tmp = reader.readLine().trim().split(" ");
n = Integer.parseInt(tmp[0]); // 采用BufferedReader读取大量数据时,大大减少了运行时间
w = Long.parseLong(tmp[1]);
String[] s = reader.readLine().trim().split(" ");
stone = new int[n];
for (int i=0;i<n;i++)
stone[i] = Integer.parseInt(s[i]);
计算前n项和:
weight = new long[n+1]; // 这里数组的长度为n+1,记录0~n的各个前n项和(共n+1个)
int pos = 1;
while (pos<=n){ // 计算各个位置的前n项和
weight[pos] = weight[pos-1] + stone[pos-1];
pos++;
}
计算k值:
若是每次让区间大小从1开始取,则复杂度会达到O(n方),这在处理大量数据是非常不理想的,因此在计算下一个的前缀和时,让区间大小从k+1开始取,节省大量时间。
int k = 0;
for (int i=1;i<=n;i++){ // 若是让k一直从1开始,复杂度为O(n方),对于蓝桥杯的大规模数据很容易超时,则每次记录k的值
for (int p=k+1;i+2*p-1<=n;p++){ // 下次从k+1开始取,这样直接清除了之后的k小于当前k的的情况,节省大量时间
lw = weight[i+p-1] - weight[i-1];
rw = weight[i+2*p-1] - weight[i+p-1];
if (lw>w||rw>w) // 若当前后有一方不满足则退出循环,选择上一次的k为最终的k
break;
k=p;
}
}
完整代码:
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
public class Main{
static int n = 0; // 石头数量
static long w = 0; // 指定重量
static int[] stone; // 各个石子的重量
static long[] weight; // 各个位置的前n项和
static long lw = 0; // 当前位置i前k个石子的总重量
static long rw = 0; // 当前位置i后k个石子的总重量
public static void main(String[] args) throws IOException {
BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));
String[] tmp = reader.readLine().trim().split(" ");
n = Integer.parseInt(tmp[0]); // 采用BufferedReader读取大量数据时,大大减少了运行时间
w = Long.parseLong(tmp[1]);
String[] s = reader.readLine().trim().split(" ");
stone = new int[n];
for (int i=0;i<n;i++)
stone[i] = Integer.parseInt(s[i]);
weight = new long[n+1]; // 这里数组的长度为n+1,记录0~n的各个前n项和(共n+1个)
int pos = 1;
while (pos<=n){ // 计算各个位置的前n项和
weight[pos] = weight[pos-1] + stone[pos-1];
pos++;
}
int k = 0;
for (int i=1;i<=n;i++){ // 若是让k一直从1开始,复杂度为O(n方),对于蓝桥杯的大规模数据很容易超时,则每次记录k的值
for (int p=k+1;i+2*p-1<=n;p++){ // 下次从k+1开始取,这样直接清除了之后的k小于当前k的的情况,节省大量时间
lw = weight[i+p-1] - weight[i-1];
rw = weight[i+2*p-1] - weight[i+p-1];
if (lw>w||rw>w) // 若当前后有一方不满足则退出循环,选择上一次的k为最终的k
break;
k=p;
}
}
System.out.println(2*k);
}
}