1044. Shopping in Mars (25)

最新推荐文章于 2020-02-07 11:44:31 发布

十觞亦不醉

最新推荐文章于 2020-02-07 11:44:31 发布

阅读量342

点赞数

CC 4.0 BY-SA版权

分类专栏： PAT(Advanced Level) 文章标签： C++ PAT Shopping in Mars

本文链接：https://blog.youkuaiyun.com/zhangpiu/article/details/50737261

PAT(Advanced Level) 专栏收录该内容

103 篇文章

订阅专栏

本文介绍了一种使用二分查找算法优化求解特定子数组和问题的方法，避免了直接双层循环带来的效率低下。通过预先计算累积和并利用二分查找定位目标区间，有效地找到了满足条件的最小子数组。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

直接双层循环会超时，后来才发现考察二分查找，钻石值存为和的形式，然后二分搜索。

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <limits>

using namespace std;

int search(int first, int& last, int m, vector<int>& sum){
	int left = first;
	int right = (int)sum.size()-1;

	while(left < right){
		const int mid = left + (right-left)/2;
		if(sum[mid] - sum[first-1] >= m) right = mid;
		else left = mid+1;
	}

	last = left;
	return sum[last] - sum[first-1];
}

int main(){
	int n, m;
	scanf("%d%d", &n, &m);

	vector<int> sum(n+1);
	for(int i = 1; i <= n; ++i){
		scanf("%d", &sum[i]);
		sum[i] += sum[i-1];
	}

	int minSum = numeric_limits<int>::max();
	vector<pair<int,int>> results, mins;

	for(int i = 1; i <= n; ++i){
		int last;
		int tmpsum = search(i, last, m, sum);

		if(tmpsum == m){
			results.push_back({i, last});
		}else if(tmpsum > m){
			if(minSum > tmpsum){
				minSum = tmpsum;
				mins = vector<pair<int,int>>(1, {i, last});
			}else if(minSum == tmpsum){
				mins.push_back({i, last});
			}
		}
	}

	if(results.empty()) results = mins;
	for(auto& result : results){
		printf("%d-%d\n", result.first, result.second);
	}

	return 0;
}