【蓝桥杯】试题 算法训练 礼物

资源限制

时间限制：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 <iostream>
using namespace std;
long long weight[1000000];
long long weightsum[1000000]; 
long long N,S;
int main()
{
  cin>>N>>S;
  for(int i=0;i<N;i++){
  	cin>>weight[i];
  }
  int k=1;
  weightsum[0]=weight[0];
  //求前缀和
  for(int i=1;i<N;i++){
  	weightsum[i]=weightsum[i-1]+weight[i];
  }
  int i=0;
  while(2*k<=N&&i<N){
  	for(i=0;i<N;i++){
  		if(i==0){
  			if((weightsum[k-1]<=S)&&(weightsum[2*k-1]-weightsum[k-1]<=S)){
  			k++;
  			break;	
  			}
  			
  		}
  		else {
  			if((i+2*k<N)&&(weightsum[i+k-1]-weightsum[i-1]<=S)&&(weightsum[i+2*k-1]-weightsum[i+k-1]<=S)){
  			k++;
  			break;	
  			}
  		}
  	}
  }
  cout<<2*(k-1)<<endl;
  return 0;
}