《近序数组》算法题

【问题描述】

    对于一个具有 n 个元素的数组，如果可以将其分为两个部分，它的各个部分都是一个非严格有序数组，则我们称这样的数组为近序数组，例如

数组 1、2、3、4、4、3、2、1是近序数组，数组 4、2、1、2、3、4也是近序数组，而1、5、7、3、9、3就不是近序数组，数组1、3、3、4是有序数组，是近序数组的特例。

【输入形式】

    输入的第一行为一个整数n，表示数组的元素个数。

    接下来的一行，表示数组的元素。

【输出形式】

如果给定的数组是近序数组输出Yes，否则输出No。

【样例输入1】
8
1 2 3 4 4 3 2 1

【样例输出1】
Yes

【样例输入2】
4
3 4 3 4

【样例输出2】
Yes

【样例输入3】
5
3 4 3 4 3

【样例输出3】
No

C++代码：

#include <iostream>
using namespace std;
int main()
{
    int n;
    int flag;
    cin >> n;
    int a[100];
    for (int i = 0; i < n; i++)
    {
        cin >> a[i];
    }

    if (a[0] < a[1]) flag = 1;
    if (a[0] > a[1]) flag = 0;
    if (a[0] == a[1]) flag = 2;
    
    int p = 1;
    while (true)
    {
        if ((p == n - 1) || (flag==1 && a[p] > a[p + 1]) || (flag==0 && a[p] < a[p + 1]))
        {
            break;
        }
        if (flag == 2)
        {
            if (a[p] < a[p+1]) flag = 1;
            if (a[p] > a[p+1]) flag = 0;
        }
        p++;
    }

    if (p != n - 1)
    {
        if (a[p+1] < a[p+2]) flag = 1;
        if (a[p+1] > a[p+2]) flag = 0;
        p++;
    }

    while (true)
    {
        if ((p == n - 1) || (flag==1 && a[p] > a[p + 1]) || (flag==0 && a[p] < a[p + 1]))
        {
            break;
        }
        if (flag == 2)
        {
            if (a[p] < a[p + 1]) flag = 1;
            if (a[p] > a[p + 1]) flag = 0;
        }
        p++;
    }
    if (p == n - 1)
    {
        cout << "Yes";
    }
    else
    {
        cout << "No";
    }
}