hdu 5532 Almost Sorted Array 2015 长春区域赛 模拟

题目

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=5532

题目来源：2015长春网络赛第二题。

简要题意：判断一数列是否能够去掉一个元素变为不递增或不递减数列。

数据范围：$1\leqslant T\leqslant 2000;\quad 2\leqslant n\leqslant 10^5;\quad 1\leqslant a_i\leqslant 10^5$

题解

题意挺好懂的，但是比较卡细节，队友都挂在这题上了。

基本的想法是处理四个数组来判断从前到某位置是否不递减，是否不递增，从后到某位置是否不递减，是否不递增。

然后对于当前的位置就只有两个情况判断下就行了。

实现

需要注意边界上的问题，这个可以统一到处理的数组里。

代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <stack>
#include <queue>
#include <string>
#include <vector>
#include <set>
#include <map>

#define pb push_back
#define mp make_pair
#define all(x) (x).begin(),(x).end()
#define sz(x) ((int)(x).size())
#define fi first
#define se second
using namespace std;
typedef long long LL;
typedef vector<int> VI;
typedef pair<int,int> PII;
LL powmod(LL a,LL b, LL MOD) {LL res=1;a%=MOD;for(;b;b>>=1){if(b&1)res=res*a%MOD;a=a*a%MOD;}return res;}
// head
const int N = 1E5+5;

int a[N];

bool lrinc[N];
bool lrdec[N];
bool rlinc[N];
bool rldec[N];

bool ck(int x, int n) {
    return (x == 1 || x == n);
}
int main() {
    int t, n, cas = 1;
    scanf("%d", &t);
    while (t--) {
        scanf("%d", &n);
        for (int i = 1; i <= n; i++) {
            scanf("%d", a+i);
        }
        lrinc[0] = lrdec[0] = lrinc[1] = lrdec[1]
        = rlinc[n+1] = rldec[n+1] = rlinc[n] = rldec[n] = true;

        for (int i = 2; i <= n; i++) {
            lrinc[i] = (a[i] >= a[i-1]) ? lrinc[i-1] : false;
            lrdec[i] = (a[i] <= a[i-1]) ? lrdec[i-1] : false;
        }
        for (int i = n-1; i > 0; i--) {
            rlinc[i] = (a[i] >= a[i+1]) ? rlinc[i+1] : false;
            rldec[i] = (a[i] <= a[i+1]) ? rldec[i+1] : false;
        }

        bool res = false;
        for (int i = 1; i <= n; i++) {
            if ((lrinc[i-1] && rldec[i+1] && (a[i-1] <= a[i+1] || ck(i, n)))
                    || (lrdec[i-1] && rlinc[i+1] && (a[i-1] >= a[i+1] || ck(i, n)))) {
                res = true;
                break;
            }
        }
        puts(res ? "YES" : "NO");
    }
    return 0;
}