蓝桥杯练习系统-算法-礼物-贪心

碎冰冰&算法入门

已于 2024-03-03 21:45:43 修改

阅读量440

点赞数 12

文章标签：蓝桥杯算法职场和发展

于 2024-03-03 21:00:43 首次发布

本文链接：https://blog.csdn.net/ha_ha_ha_wys/article/details/136437941

版权

文章讲述了JiaoShou在旅行中遇到的问题，需按特殊规则取石子作为礼物。通过计算前缀和并动态调整取石子的范围K，找出最多能带走的石子数量。问题涉及C++编程和优化算法解决。

摘要由CSDN通过智能技术生成

试题算法训练礼物

内存限制：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个1即可。

数据规模和约定

　　对于20%的数据：N<=1000
　　对于70%的数据：N<=100,000
　　对于100%的数据：N<=1000,000，S<=10^12，每个石子的重量小于等于10^9，且非负

如果取K=N/2，然后递减扫描，暴力求解，即时使用前缀和也只能通过20%的数据，所以暴力求解行不通。

首先，输入数据，然后求出前缀和，从头到尾，在每个点向两边扩展，K的初始值为0，判断扩展是否可行，可以则K++，下一个点直接从K的当前值开始（不用再从0开始），最后成功通过所有数据，详细代码如下。

#include<iostream>

using namespace std;

typedef long long ll;

ll arr[1000005]={0};
int N,K=0;
bool check(int left,int mid,int right);
ll S;
bool flag = false;

int main(){
	cin>>N>>S;
	for(int i=1;i<=N;i++){
		cin>>arr[i];
		arr[i] += arr[i-1];
	}
	for(int i=1;i<=N;i++){
		while(i-K>=0 && i+K<=N && check(i-K,i,i+K)) K++;
	}
	cout<<2*(K-1);
	return 0;
}

bool check(int left,int mid,int right){
	ll sumL=arr[mid]-arr[left],sumR=arr[right]-arr[mid];
	if(sumL<=S && sumR<=S) return true;
	return false;
}