蓝桥杯练习系统（算法训练）ALGO-1003 礼物

资源限制

内存限制：256.0MB   C/C++时间限制：1.0s   Java时间限制：3.0s   Python时间限制：5.0s

问题描述

　　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;

const int N=10e6+10;
int a[N];
long long b[N];//前缀和 
int n;
long long s;

bool check(int mid){
	for(int i=mid;i<=n-mid;i++){
		if(b[i]-b[i-mid]<=s&&b[i+mid]-b[i]<=s){
			return true;
		}
	}
	return false;
}
int main(){
	scanf("%d%lld",&n,&s);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		b[i]=b[i-1]+a[i];
	}
	int l=1,r=n;
	while(l<=r){
		int mid=(l+r)>>1;
		if(check(mid)){
			l=mid+1;
		}else{
			r=mid-1;//如果不符合，结果r减1，即k减1 
		}
	}
	printf("%d\n",2*r);
} 

 

思路：使用二分算法。这里的mid表示的是k，找到合适的mid后，将它放在r中保存，所以最终的答案为2*r。注意数据范围。

以题目所给的样例为例：

在最开始时，一共有8个石子，什么都不管，最长的长度为mid*2=8，但是并不符合题意，因此mid需要减1，将其保存在r中，此时r=3，目前什么都不管，最长的长度为r*2=6，发现符合题意，即最后的答案为6。

（后面的可以不看）

一开始并没有想到二分，直接暴力，结果只有20分，后面全都超时了……

#include<iostream>
#include<math.h>
using namespace std;
int n;
long long s;
const int N=10e6+10;
int a[N];
long long int b[N];//前缀和 
bool two_sum(int begin,int k){
	if(b[begin+k]-b[begin]<=s&&b[begin+2*k]-b[begin+k]<=s)
	return true;
	else return false;
}
int main(){
	scanf("%d%lld",&n,&s);
	for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
		b[i]=b[i-1]+a[i]; 
	}
	int ans=0,res=0;
	for(int i=1;i<=n;i++){//从第a[i]个开始拿 
		//拿2*k个
		for(int k=(n-i+1)/2;k>0;k--) {
			long long int sum=0;
			if(two_sum(i,k)){
				res=2*k;
				break;
			}
		}
		ans=max(ans,res);
	}
	printf("%d\n",ans);
	return 0;
} 

 

从第a[i]个开始拿（下标从1开始，为了方便求前缀和），最多可以拿（n-i+1）个，如果不符合的话，拿（n-i+1）- 2个，还不符合的话，拿（n-i+1）-4个，直到符合题意，最终的个数即为从第a[i]个开始拿的最大值res。每个从第a[i]开始拿的情况都有一个最大值。

比较每个最大值，取最终的最大值。