第3题 选蛋糕 查看测评数据信息
小明最近爱上了做奶油小蛋糕。一天他在家中做了一排共n个蛋糕,第i个蛋糕的大小为ai 。制作过程中,通过比较相邻两个蛋糕,小明确保相邻蛋糕的大小差值不超过1,即任意第i个蛋糕与相邻蛋糕之间存在|ai-ai-1|≤1并且|ai-ai+1|≤1的关系。小明欣赏了一下自己的作品,他决定在这一排蛋糕中只选取一段摆放在一起的蛋糕(连续),带给班级的同学分享。为了公平,要求这一段蛋糕中,任意两蛋糕大小差距不超过1。请你帮助小明求出最多能选出多少个蛋糕。
输入格式
第一行为正整数t(≤5),表示数据组数;每组数据中,第一行为蛋糕个数正整数n(≤100000),第二行为n个蛋糕的大小正整数ai(≤100000)。
输出格式
对于每组数据,输出最多能选出多少个蛋糕。
输入/输出例子1
输入:
2
5
1 2 3 3 2
14
20 19 20 21 20 19 19 18 17 17 18 18 19 18
输出:
4
5
样例解释
无
下面是程序
#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n, m, a[100000 + 10],mi=1e18,b[100000+10],f1;
ll check(ll x) {
if (x == 1)return true;
for (ll i = 1; i <= x; i++) b[i] = a[i];
sort(b + 1, b + m + 1);
f1 = lower_bound(a + 1, a + x + 1, b[1] + 2) - a;
if (f1 > x) return true;
for (ll i = x + 1; i <= m; i++) {
b[(i-1)%x+1] = a[i];
sort(b + 1, b + x + 1);
f1 = lower_bound(a + 1, a + m + 1, b[1] + 2) - a;
cout<<f1<<endl;
if (f1 > i+x || f1 < i)return true;
}
return false;
}
int main() {
cin >> n;
while (n--) {
cin >> m;
mi = 1e18;
for (ll i = 1; i <= m; i++) {
cin >> a[i];
if (i > 1)mi = min(mi, a[i] - a[i - 1]);
}
a[m+1]=1000000+1;
if (mi > 1) {
cout << 1 << endl;
continue;
}
ll l = 0, r = m + 1;
while (l + 1 < r) {
ll mid = (l + r) >> 1;
if (check(mid))l = mid;
else r = mid;
}
cout << l << endl;
}
return 0;
}