1402 区间取数2

// 1402 区间取数2.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
//

/*

http://oj.daimayuan.top/course/22/problem/1090

给你 n个数 a1,a2,...,an 和一个整数 k
，你需要在这 n 个数中选出连续一段数，使得这些数的和不超过 k。

请问最多能选几个数。

输入格式
输入第一行两个整数 n,k。

接下来一行 n 个数 a1,a2,...,an。

输出格式
输出一行一个数表示答案。

样例输入
4 4
2 3 3 3
样例输出
1
数据范围
对于 100% 的数据，保证 1≤n≤106,1≤k≤109,1≤ai≤104。
*/
#include <iostream>
#include <deque>


using namespace std;

const int N = 1000010;
int a[N];

int main()
{
	int n,k; cin >> n >>k;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}

	int l = 1; int r = 1;
	long long sum = a[1];
	int ans = 0;
	while (l <= n && r <= n) {
		if (sum <= k) {
			ans = max(ans, r - l + 1);
			r++; sum += a[r];
		}
		else {
			if (l < r) {
				sum -= a[l];
				l++;
			}
			else {
				l = l + 1; r = l;
				sum = a[l];
			}
		}
	}

	cout << ans << endl;


	return 0;
}
 

使用二分逐个尝试合适的长度，前缀和计算得到最长多少连续数的和小于等于k

#include <iostream>
#include <deque>

using namespace std;

const int N = 1000010;
int a[N];
long long pre[N];
int n, k;

bool check(int l) {
	for (int i = 0; i <= n; i++) {
		int r = i + l;
		if (r <= n && pre[r] - pre[i] <= k) {
			return true;
		}
	}

	return false;
}


int main()
{
	cin >> n >>k;
	for (int i = 1; i <= n; i++) {
		cin >> a[i]; pre[i] = pre[i - 1] + a[i];
	}

	int l = 0; int r = n;
	while (l < r) {
		int mid = (l + r + 1) >> 1; 
		if (check(mid)) {
			l = mid;
		}
		else {
			r = mid - 1;
		}
	}
 
	cout << l << endl;

	return 0;
}