题目如下:
小A手上有一串珍珠链,共有 N 颗珍珠,珍珠的颜色值用 1 到 M 的整数表示。我们称一个珍珠链的 2×k 长度的子串是“漂亮的”,当且仅当该子串中前 k 个珠子的颜色值之和或最后 k 个珠子的颜色值之和都小于等于 S。
现给出珍珠链每颗珠子的颜色值,对于每一颗珠子,输出从该珍珠开始最长的漂亮子串的长度。
第一行包含整数 N 和 S。
下面的 N 行,每行包含珍珠链中的一个颜色值 si。这些整数都是正的且它们的和不超过 2×109。
输出共 N 行。第 i 行包含一个整数,表示从第 i 个珍珠开始最长的漂亮子串的长度
如果当前位置上没有漂亮子串,输出 0。
5 10000
1
1
1
1
1
4
4
2
2
0
5 9
1
1
10
1
9
2
0
0
2
0
8 3
1
1
1
1
1
1
1
1
6
6
6
4
4
2
2
0
【样例解释#1】
对于样例 1 的第一个位置,k 的值最大为 2,并且前两个珍珠颜色值加起来为 2,后两个珍珠颜色值加起来为 2,都小于 10000,故最长长度的漂亮子串长度为 4。
【数据范围】
对于 100% 的数据,2≤N≤105,1≤S≤2×109。 我的代码是这样的:
#include<bits/stdc++.h>
using namespace std;
int n,s,ans;
int arr[200010],b[200010];
int check(int a,int x){
if(b[a+x/2]-b[a]<=s&&b[a+x]-b[a+x/2]<=s){
return 1;
}
else{
return 0;
}
}
int main(){
cin>>n>>s;
for(int i=1;i<=n;i++){
cin>>arr[i];
b[i]=arr[i]+b[i-1];
}
int l=0,r=0;
for(int i=1;i<=n;i++){
cout<<endl<<endl<<endl;
r=n-i;
l=0;
ans=0;
while(l<=r){
int mid=(l+r)/2;
if(mid%2==1){
r--;
continue;
}
if(check(i,mid)){
ans=mid;
l=mid+1;
}
else{
r=mid-1;
}
}
cout<<ans<<endl;
}
return 0;
}