参考题解
https://www.acwing.com/solution/content/59384/
作者:wxcddszq。
题目描述
给定一个长度为 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 两两不同。
AC Code
#include <algorithm>
#include <iostream>
#include <vector>
#include <string>
using namespace std;
int main()
{
int n;
cin>>n;
vector<int> v(n);
for(int i=0;i<n;i++){
cin>>v[i];
}
string s;
cin>>s;
int l=0,r=0;
while(r < n){
while(s[r] == '1') r++;
sort(v.begin()+l,v.begin()+r+1);
r ++;
l = r;
}
if(is_sorted(v.begin(),v.end())){
cout<<"YES";
}
else{
cout<<"NO";
}
return 0;
}
解释
〇由于这是一道简单题,思路与代码几乎重合,因此上述的代码与题解作者的代码无太大差异,但的确是出自本人之手,而不是简单的复制粘贴,特此声明。
①此处使用了vector数据结构作为存储输入数据的数组,不管是在AcWing上刷题还是在CodeForces中刷题,我发现高手都偏爱使用vector作为数组。以我目前的经验,使用vector有以下好处:
- 在sort排序时可以直接使用vector迭代器的begin和end作为开始和结束,而不需要给出排序的空间,十分便利。
- 有一些题目是需要我们不断地记录中间值(比如搜索时,最后输出一下到达终点的路径),这时候使用vector的push_back方法是十分简洁的,而使用vector的size方法就可以得到数组的大小,即走了几步,很便捷。
- vector可以存储如pair<=,=>这种复杂的数据结构。
因此,以后也要养成使用vector的好习惯。
②显然,对于一段连续的1,这一段当中的顺序是可以随意改变的,因此,直接使这一段为升序即可。此处使用双指针记录哪一段可以进行排序,如果这些段排好序后是升序的,那么输出yes,否则no。
③可以使用is_sorted(v.begin(),v.end())
直接判断整个数组是否已经排好序,返回值为boolean。