#include<bits/stdc++.h>
using namespace std;
#define int long long
signed main(){
ios::sync_with_stdio(0);
cin.tie(0);
cout.tie(0);
int t;
cin>>t;
while(t--){
int n;
cin>>n;
string s;
cin>>s;
int sum=0;
if(s[0]=='1') sum++;
for(int i=1;i<n;i++){
if(s[i]!=s[i-1]) sum++;
}if(sum==2){
sum-=1;
}
else if(sum>2){
sum-=2;
}
int ans=n+sum;
cout<<ans<<"\n";
}
}
有注释版:
#include<bits/stdc++.h> // 引入所有标准库头文件,常用于竞赛中简化代码
using namespace std;
#define int long long // 将int重定义为long long,防止整数溢出
signed main(){
ios::sync_with_stdio(0); // 关闭C++与C标准流的同步,加快输入输出速度
cin.tie(0); // 解除 cin 与 cout 的绑定,提高效率
cout.tie(0);
int t; // 表示测试用例的数量
cin >> t;
while(t--){ // 对每个测试用例进行处理
int n; // 二进制字符串的长度
cin >> n;
string s; // 输入的二进制字符串
cin >> s;
int sum = 0; // 用来统计相邻字符变化的次数
if(s[0] == '1') sum++; // 如果第一个字符是'1',因为初始手指在'0'按钮上,需要切换一次
for(int i = 1; i < n; i++){
// 如果当前位置的字符和前一个不同,表示需要切换按钮
if(s[i] != s[i-1]) sum++;
}
// 下面是关键优化逻辑:
// 如果整个字符串中正好有2次变化(如 '000111'),
// 通过反转中间子串,可以减少1次切换
if(sum == 2){
sum -= 1;
}
// 如果变化次数大于2(如 '010101'),可以通过反转一个合适的区间减少2次切换
else if(sum > 2){
sum -= 2;
}
// 操作总数 = 按键次数(每个字符都要按) + 切换次数
int ans = n + sum;
cout << ans << "\n"; // 输出结果
}
}