题意:给出k (4≤k≤5000) 个互不相同的整数组成的序列 Ni ,判断是否存在4个整数 Np,Nq,Nr和Ns(1≤p<q<r<s≤k) ,使得 Nq>Ns>Np>Nr 或者 (Nq<Ns<Np<Nr) 。(本段摘自《算法竞赛入门经典(第2版)》
分析:
首先预处理maxm[i][j],表示从i开始到j的最大值,然后再处理vec[i],表示从i + 1开始所有大于a[i]的数,并排序。先枚举p和r,在根据maxm[i][j]直接选取最大的
Nq
,最后通过二分在vec[j]中查找符合条件的
Ns
,如果找到直接返回true,否则返回false。对输入的数列正序和逆序都处理一遍即可。
代码:
#include <iostream>
#include <algorithm>
#include <fstream>
#include <cstring>
#include <vector>
#include <queue>
#include <cmath>
#include <stack>
using namespace std;
const int maxn = 5000 + 5;
int T, n, tmp, pos;
int a[maxn], b[maxn], maxm[maxn][maxn];
vector< int > vec[maxn];
bool judge(int x[])
{
for (int i = 0; i < n; ++i)
{
tmp = x[i];
maxm[i][i] = x[i];
for (int j = i + 1; j < n; ++j)
if (x[j] > tmp)
{
maxm[i][j] = x[j];
tmp = x[j];
}
else
maxm[i][j] = maxm[i][j - 1];
}
for (int i = 0; i < n; ++i)
{
vec[i].clear();
for (int j = i + 1; j < n; ++j)
if (x[j] > x[i])
vec[i].push_back(x[j]);
sort(vec[i].begin(), vec[i].end());
}
for (int i = 0; i < n; ++i)
for (int j = i + 2; j < n; ++j)
if (x[j] < x[i])
if (maxm[i][j - 1] > x[j])
{
pos = lower_bound(vec[j].begin(), vec[j].end(), x[i]) - vec[j].begin();
if (pos != vec[j].size())
if (vec[j][pos] < maxm[i][j - 1])
return true;
}
return false;
}
int main()
{
scanf("%d", &T);
for (int C = 0; C < T; ++C)
{
scanf("%d", &n);
for (int i = 0; i < n; ++i)
{
scanf("%d", &a[i]);
b[n - i - 1] = a[i];
}
if (judge(a) || judge(b))
printf("YES\n");
else
printf("NO\n");
}
return 0;
}