920C - Swap Adjacent Elements - CodeForces Solution

#include<iostream>
#include<string.h>
using namespace std;
int sum[200005];
int a[200005];
int main(){
	int n;
	cin >> n;
	for(int i = 1; i <= n; i ++){
		scanf("%d",&a[i]);
	}
	char s[200005];
	scanf("%s",s + 1);
	int l = strlen(s + 1);
	for(int i = 1; i <= l; i ++){
		sum[i] = sum[i - 1] + s[i] - '0';
	}
	for(int i = 1; i <= n; i ++){
		if(a[i] > i){
			if(sum[a[i] - 1] - sum[i - 1] == (a[i] - i))
				continue;
			else{
				puts("NO");
				return 0;
			}
		}
		if(a[i] < i){
			if(sum[i - 1] - sum[a[i] - 1] == (i - a[i]))
				continue;
							else{
								puts("NO");
								return 0;
							}
		}
		
	}
	puts("YES");
	return 0;
}