吉哥系列故事——完美队形II
吉哥又想出了一个新的完美队形游戏!
假设有n个人按顺序站在他的面前,他们的身高分别是h[1], h[2] ... h[n],吉哥希望从中挑出一些人,让这些人形成一个新的队形,新的队形若满足以下三点要求,则就是新的完美队形:
1、挑出的人保持原队形的相对顺序不变,且必须都是在原队形中连续的;
2、左右对称,假设有m个人形成新的队形,则第1个人和第m个人身高相同,第2个人和第m-1个人身高相同,依此类推,当然如果m是奇数,中间那个人可以任意;
3、从左到中间那个人,身高需保证不下降,如果用H表示新队形的高度,则H[1] <= H[2] <= H[3] .... <= H[mid]。
现在吉哥想知道:最多能选出多少人组成新的完美队形呢?
Input
输入数据第一行包含一个整数T,表示总共有T组测试数据(T <= 20);
每组数据首先是一个整数n(1 <= n <= 100000),表示原先队形的人数,接下来一行输入n个整数,表示原队形从左到右站的人的身高(50 <= h <= 250,不排除特别矮小和高大的)。
Output
请输出能组成完美队形的最多人数,每组输出占一行。
Sample Input
2 3 51 52 51 4 51 52 52 51
Sample Output
3 4
#include <iostream>
#include <cstring>
#include <algorithm>
#include <string>
#include <cstdio>
#include <cmath>
using namespace std;
const int N = 300007;
int s[N], p[N];
int main() {
int t;
scanf ("%d", &t);
while (t--) {
memset(p, 0, sizeof(p));
int n;
scanf ("%d", &n);
int k = 0;
s[k++] = -10;
s[k++] = -1;
for (int i = 1; i <= n; ++i) {
scanf ("%d", &s[k++]);
s[k++] = -1;
}
s[k++] = -5;
int mx = 0, len = 0, id = 0;
for (int i = 1; i < k; ++i) {
p[i] = mx > i ? min(p[2*id - i], mx - i) : 1;
while (s[i + p[i]] == s[i - p[i]]
&& s[i + p[i]] <= s[i + p[i]-2]) ++p[i]; //注意-1时的情况
if (mx < i + p[i]) mx = i + p[i], id = i;
if (len < p[i] - 1) len = p[i] - 1;
}
printf ("%d\n", len);
}
return 0;
}