1739E - Cleaning Robot - CodeForces Solution

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int mod=1e9+7;
const int maxn=2e5+10;
int a[maxn][2],dp[maxn][2];
int main()
{
	ios::sync_with_stdio(false);
	cin.tie(nullptr);
	int n;
	cin>>n;
	for(int i=1;i<=n;i++){
		char x;
		cin>>x;
		if(x=='0'){
			a[i][0]=0;
		}
		else{
			a[i][0]=1;
		}
	}
	for(int i=1;i<=n;i++){
		char x;
		cin>>x;
		if(x=='0'){
			a[i][1]=0;
		}
		else{
			a[i][1]=1;
		}
	}
	memset(dp,-0x3f,sizeof(dp));
	dp[1][0]=0;
	for(int i=1;i<=n;i++){
		for(int j=0;j<2;j++){
			dp[i+1][j]=max(dp[i+1][j],dp[i][j]+a[i+1][j]);
			if(a[i][j^1]){
				dp[i+2][j^1]=max(dp[i+2][j^1],dp[i][j]+a[i][j^1]+a[i+1][j^1]+a[i+2][j^1]);
			}
			//cerr<<i+2<<' '<<(j^1)<<' '<<dp[i+1][j]<<' '<<dp[i+2][j^1]<<'\n'; 
		}
	}
	int ans=0;
	for(int i=n+1;i<=n+2;i++){
		for(int j=0;j<2;j++){
			ans=max(ans,dp[i][j]);
		}
	}
	cout<<ans<<'\n';
	return 0;
}