代码会出现超时,不过思路值得借鉴:
输入a[1]~a[n],sum[i]=a[1]+a[2]+a[3]+…+a[i],由于a[]都是正数,因此sum[]是递增序列,同时一个技巧点,在于sum[0]=0,这样sum[i]-sum[0]=a[i].
问题等价于枚举左端点i,在sum数组的[i,n]内查找值为sum[i-1]+s的元素
用二分法找到第一个大于等于sum[i-1]+S
#include <iostream>
#include <cstdio>
#include <string.h>
#include <vector>
#include <algorithm>
#include <cmath>
using namespace std;
const int maxn=100010;
int nearS=100000010;
int sum[maxn]={0},n,m,a[maxn];
//找到第一个大于等于x的元素位置
int lower_bound(int A[],int left,int right,int x){
int mid;
while(left<right){
mid=(left+right)/2;
if(A[mid]>=x){
right=mid;
}else{
left=mid+1;
}
}
return left;
}
int main(void){
scanf("%d%d",&n,&m);
int temp=0;
for(int i=1;i<=n;i++){//下标从1开始,sum[0]取0
scanf("%d",&a[i]);
// sum[i]+=a[i];
temp+=a[i];
sum[i]=temp;//sum是递增函数
}
int l,min=0;
for(int i=1;i<=n;i++){
// l=lower_bound(sum,i,n,sum[i-1]+m);
// if(l!=0){
// break;
// }
int j=lower_bound(sum,i,n+1,sum[i-1]+m);
if(sum[j]-sum[i-1]==m){
nearS=m;
break;
}else if(sum[j]-sum[i-1]<nearS&&j<=n){//不断更新直到nearS最小
nearS=sum[j]-sum[i-1];
}
}
// nearS=sum[l];
for(int i=1;i<=n;i++){
for(int j=i+1;j<=n+1;j++){
if(sum[j-1]-sum[i-1]==nearS){
printf("%d-%d\n",i,j-1);
}
}
}
}