Codeforces Round #807 (Div. 2)

比赛链接：

https://codeforces.com/contest/1705

D. Mark and Lightbulbs

题意：

给定两个字符串 \(s\) 和 \(t\)，要求将 \(s\) 变成 \(t\)。每次可以选择一个满足 \(s_{i - 1} != s_{i + 1}\) 且 \(2 <= i <= n - 1\) 的下标 \(i\)，让 \(s_i\) 翻转，即从 0 变成 1 或者从 1 变成 0。问最少多少步可以实现字符串的转化，若不能，输出 -1。

思路：

容易发现，变化的过程其实就是让某段 1 变长或者变短，但不会变没掉。所以只需要计算两个字符串中 1 的段的数量，如果相同，就有答案，答案为字符串中对应的每段 1 的开头和结尾的坐标差值之和。

代码：

#include <bits/stdc++.h>
using namespace std;
#define LL long long
void solve(){
	LL n;
	cin >> n;
	string s, t;
	cin >> s >> t;
	if (s[0] != t[0] || s[n - 1] != t[n - 1]){
		cout << "-1\n";
		return;
	}
	vector <LL> a, b;
	for (int i = 0; i < n - 1; i ++ ){
		if (s[i] != s[i + 1]) a.push_back(i);
		if (t[i] != t[i + 1]) b.push_back(i);
	}
	if (a.size() != b.size()){
		cout << "-1\n";
		return;
	}
	LL ans = 0;
	for (int i = 0; i < (int)a.size(); i ++ ){
		ans += abs(a[i] - b[i]);
	}
	cout << ans << "\n";
}
int main(){
	ios::sync_with_stdio(false);cin.tie(0);
	LL T;
	cin >> T;
	while(T -- ){
		solve();
	}
	return 0;
}