题目描述
东东和他的女朋友去寿司店吃晚餐,他发现了一个有趣的事情,这家餐厅提供的 n 个的寿司被连续的放置在桌子上 (有序),东东可以选择一段连续的寿司来吃
东东想吃鳗鱼,但是东妹想吃金枪鱼。核 平 起 见,他们想选择一段连续的寿司(这段寿司必须满足金枪鱼的数量等于鳗鱼的数量,且前一半全是一种,后一半全是另外一种)我们用1代表鳗鱼,2代表金枪鱼。
比如,[2,2,2,1,1,1]这段序列是合法的,[1,2,1,2,1,2]是非法的。因为它不满足第二个要求。
东东希望你能帮助他找到最长的一段合法寿司,以便自己能吃饱。
输入格式
第一行:一个整数n(2≤n≤100000),寿司序列的长度。
第二行:n个整数(每个整数不是1就是2,意义如上所述)
输出格式
一个整数(代表东东可以选择的最长的一段连续的且合法的寿司)
样例
Input
7
2 2 2 1 1 2 2
Output
4
Input
6
1 2 1 2 1 2
Output
2
Input
9
2 2 1 1 1 2 2 2 2
Output
6
解题思路
写这道题时我饿了。
思路很明确,扫描一遍寿司序列:
对于某一个寿司,判断它和前一个寿司是否相同:
- 相同,则此类寿司的当前寿司数++
- 不同,则此类首次的当前寿司书=1(即重新计算了)
记录每个寿司位置对应的金枪鱼寿司的数量和鳗鱼寿司的数量的较小值,答案即为这些最小值里最大的那个,乘以2(毕竟两种寿司都要吃)。
这样的时间复杂度就很可观了,不会超时。
详见代码:
#include <iostream>
#include <cstring>
#include <algorithm>
const int N = 100002;
using namespace std;
int a[N];
int main()
{
int n;
cin >> n;
for (int i = 0; i < n; i++)
cin >> a[i];
int sushi[3];
memset(sushi, 0, sizeof sushi);
int ans = 0;
if (a[0] == 1)
{
sushi[1]++;
}
else
{
sushi[2]++;
}
for (int i = 1; i < n; i++)
{
if (a[i] == a[i - 1])
{
sushi[a[i]]++;
}
else
{
sushi[a[i]] = 1;
}
ans = max(ans, min(sushi[1], sushi[2]));
}
cout << ans * 2 << endl;
}