资源限制
时间限制: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;
}