PAT1044. Shopping in Mars (25)(二分)

题意:

给出一系列数字,要求输出子序列使其和等于m,如果没有等于m的子序列就输出和减去m的差最小的子序列

要点:

这题的时间要求是100ms,n的量级是10^5,O(n^2)是肯定不行了,但我一开始算了一下O(nlogn)觉得不大行,就一直在想O(n)的方法,实际上O(nlogn)=10^5*17=1.7*10^6差不多是行的。接下来二分就行了。

#include<iostream>
#include<string>
#include<vector>
#include<map>
#include<set>
#include<stack>
#include<sstream>
#include<functional>
#include<algorithm>
using namespace std;
const int INF = 0xfffff;
const int maxn = 100050;
int num[maxn],sum[maxn];
vector<pair<int, int>> res;
int n, m;

void binery(int i, int &j, int &tempMin) {
	int left = i, right = n;
	while (left < right) {
		int mid = (left + right) / 2;
		if (sum[mid] - sum[i - 1] < m) {
			left = mid+1;
		} else
			right = mid;
	}
	j = right;
	tempMin = sum[j] - sum[i - 1];
}

int main() {
	scanf("%d%d", &n,&m);
	for (int i = 1; i <= n; i++) {
		scanf("%d", &num[i]);
		sum[i] = sum[i - 1] + num[i];
	}
	int Min = sum[n];
	for (int i = 1; i <= n; i++) {
		int j, tempMin;
		binery(i, j, tempMin);
		//cout << j << " " << tempMin << endl;
		if (tempMin >= m) {
			if (Min > tempMin) {
				Min = tempMin;
				res.clear();
				res.push_back({ i,j });
			} else if (Min == tempMin) {
				res.push_back({ i,j });
			}
		}
	}
	for (auto x : res) {
		printf("%d-%d\n", x.first, x.second);
	}
	return 0;
}

posted @ 2018-04-28 16:02  seasonal  阅读(76)  评论(0编辑  收藏  举报