1. 问题描述:
给定一个长度为 n 的数组 a1,a2,…,an。该数组是一个 1∼n 的排列。数组的前 n−1 个位置中,部分位置可以进行交换操作,将该位置的元素与后面相邻位置的元素进行互换。交换操作的次序和次数均不限。请你判断给定的数组能否通过交换操作变为一个升序数组。
输入格式
第一行包含整数 n。第二行包含 n 个整数 a1,a2,…,an。第三行包含一个长度为 n−1 的 01 字符串,第 i 个字符为 1 表示第 i 个位置可以进行交换操作,第 i 个字符为 0 表示第 i 个位置无法进行交换操作。
输出格式
给定的数组可以通过交换操作变为一个升序数组则输出 YES,否则输出 NO。
数据范围
2 ≤ n ≤ 2 × 10 ^ 5,
1 ≤ ai ≤ n,ai 两两不同。
输入样例1:
6
1 2 5 3 4 6
01110
输出样例1:
YES
输入样例2:
6
1 2 5 3 4 6
01010
输出样例2:
NO
来源:https://www.acwing.com/problem/content/3787/
2. 思路分析:
分析题目可以知道数组a是一个1~n的排列,可以发现如果当前存在一段连续的1的时候,那么内部的元素是可以相互交换直到内部元素有序的类似于冒泡排序,由于最终需要满足升序所以每一段区间对应的元素是确定的,例如对于区间[3,5]内部的元素可以相互交换那么最终需要满足区间中的元素为[3,4,5]这样才满足要求,如何判断当前区间的中的元素是否全部在当前的区间呢?由于ai两两不同所以判断区间中的最大值和最小值是否是区间端点即可,并且进一步可以发现只需要判断区间中的最大值是否是区间终点即可,因为在枚举的时候当前区间的最小值一定存在,所以只需要枚举区间端点是否是等于当前ai的最大值即可。
3. 代码如下:
class Solution:
def process(self):
n = int(input())
a = list(map(int, input().split()))
s = input()
t = 0
flag = 1
for i in range(1, n):
t = max(t, a[i - 1])
if t != i:
if s[i - 1] == "0" and t != i:
flag = 0
break
return "YES" if flag else "NO"
if __name__ == '__main__':
print(Solution().process())