3957 子序列（思维题、枚举 + 前缀最值）

1. 问题描述：

我们称一个序列是有序的，如果该序列是非严格单调递增序列或非严格单调递减序列。现在，给定一个 n 个整数的序列 a1，a2，…，an。请你找到该序列的最短非有序子序列。注意，子序列不一定连续。

输入格式

第一行包含整数 n。第二行包含 n 个整数 a1，a2，…，an。

输出格式

如果不存在非有序子序列，则输出一行 0。否则，首先在第一行输出一个整数，表示最短非有序子序列的长度，随后在第二行按顺序输出该子序列的各元素下标。如果方案不唯一，输出任意合理方案均可。

数据范围

前 6 个测试点满足 1 ≤ n ≤ 10,
所有测试点满足 1 ≤ n ≤ 10 ^ 5，−10 ^ 6 ≤ ai ≤ 10 ^ 6。

输入样例1：

5
67 499 600 42 23

输出样例1：

3
1 3 5

输入样例2：

3
1 2 3

输出样例2：

0

输入样例3：

3
2 3 1

输出样例3：

3
1 2 3
来源：https://www.acwing.com/problem/content/3960/

2. 思路分析：

分析题目可以知道非有序子序列为"V"型或者是倒"V"型，如果中间存在相等的部分我们可以去除掉这些相等的部分只保留一个点即可，而"V"字型或者倒"V"字型最少可以由三个点构成，所以原问题等价于我们能否找到三个点构成"V"字型或者倒"V"字型，如果存在那么长度一定是3，不存在则输出0；我们可以使用四个数组lmax，lmin，rmax，rmin来维护左边最大值，左边最小值，右边最大值，右边最小值对应的下标，lmax[i]表示下标从1到i对应最大值的下标，rmax[i]表示下标从i到n对应最大值的下标，lmin[i]和rmin依次类推，然后我们可以枚举每一个位置判断是否存在"V"字型或者是倒"V"字型，存在则输出对应下标即可。

3. 代码如下：

class Solution:
    def process(self):
        n = int(input())
        a = [0] + list(map(int, input().split()))
        # lmin, rmin, lmax, rmax分别存储左边最小值, 右边最小值, 左边最大值, 右边最大值对应的下标
        lmin, rmin, lmax, rmax = [0] * (n + 10), [0] * (n + 10), [0] * (n + 10), [0] * (n + 10)
        # 对于第一个位置特殊处理
        lmin[1] = lmax[1] = 1
        for i in range(2, n + 1):
            if a[lmin[i - 1]] > a[i]:
                lmin[i] = i
            else:
                lmin[i] = lmin[i - 1]
            if a[lmax[i - 1]] < a[i]:
                lmax[i] = i
            else:
                lmax[i] = lmax[i - 1]
        # 对于最后一个位置特殊处理
        rmin[n] = rmax[n] = n
        for i in range(n - 1, 0, -1):
            if a[rmin[i + 1]] > a[i]:
                rmin[i] = i
            else:
                rmin[i] = rmin[i + 1]
            if a[rmax[i + 1]] < a[i]:
                rmax[i] = i
            else:
                rmax[i] = rmax[i + 1]
        for i in range(2, n):
            if a[lmax[i - 1]] > a[i] and a[rmax[i + 1]] > a[i]:
                print("3\n{:} {:} {:}".format(lmax[i - 1], i, rmax[i + 1]))
                return
            if a[lmin[i - 1]] < a[i] and a[rmin[i + 1]] < a[i]:
                print("3\n{:} {:} {:}".format(lmin[i - 1], i, rmin[i + 1]))
                return
        print(0)


if __name__ == '__main__':
    Solution().process()