吉哥系列故事——完美隊形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;
}