题目链接:https://codeforces.com/contest/1504/problem/B
题目大意:给两个长度相等由01构成的字符串a和b问能不能将a的前缀通过有限次特定变化把a变成b,(特定的变化为0变成1,1变成0).
思路:如果没有限制条件前缀的0和1的数量必须相等,那一定可以通过不断的变化将a变成b,但要求0和1的数量必须相等,所以只需要找出由哪些地方一定要的前缀一定要发生变化,并判断该处前缀的0和1的数量是否相等即可,因为整体前缀变化就意味着相邻的两个字符的相对关系是一直保持不变的,原来是相同,前缀变化以后还是相同的,所以显然判断条件就是对于第 i ii 个字符,若 a [ i ] a[i]a[i] 、 a [ i + 1 ] a[i + 1]a[i+1] 的关系(相同还是不同),与 b [ i ] b[i]b[i] 、 b [ i + 1 ] b[i + 1]b[i+1] 的关系是相反的,那么从 i ii 开始的前缀就必须发生变化,但要注意字符串最后一位是否相等,这里我们可以在a和b的最后加上一个相同的字符(0或1)来处理
下面看代码:
#include<iostream>
using namespace std;
void solve()
{
string a,b;
int n,cnt=0;
cin>>n>>a>>b;
a.push_back('0');
b.push_back('0');
for(int i=0;i<n;i++)
{
cnt=cnt+(a[i]=='1')-(a[i]=='0');
if((a[i]==a[i+1])!=(b[i]==b[i+1])&&cnt!=0)
{
cout<<"NO"<<endl;
return;
}
}
cout<<"YES"<<endl;
}
int main()
{
int t;
cin>>t;
while(t--)
{
solve();
}
return 0;
}