题目大意:
t 组输入,每次给一个长度为 n 的 01 串,有两个操作 01-》1,或者 10-》0,问 给定的01串中有多少子串可以经过变换变成长度为1的字符串
思路分析:
这题一看数组范围,只够 for 循环走一遍 ,所以很容易想到动态规划
AC 代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int N = 2e5+10;
int t,n;
string s;
int f[N][4]={0};
void solve(){
cin>>n;
cin>>s;
s=" "+s;
ll ans=0;
f[0][0]=f[0][1]=f[0][2]=f[0][3]=0;
for(int i=1;i<=n;i++){
f[i][0]=f[i][1]=f[i][2]=f[i][3]=0;//初始化
if(s[i]=='0'){
f[i][0]=1+f[i-1][1]+f[i-1][3];
f[i][2]=f[i-1][0]+f[i-1][2];
}
else if(s[i]=='1'){
f[i][1]=1+f[i-1][0]+f[i-1][2];
f[i][3]=f[i-1][1]+f[i-1][3];
}
ans+=f[i][0]+f[i][1];
}
cout<<ans<<endl;
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0),cout.tie(0);
cin>>t;
while(t--){
solve();
}
return 0;
}
但是动态规划刚开始肯定不好想,我就没想到,看的别人的代码才想到动态规划数组存的东西和动态转移方程(手动狗头),所以我想的是正难则反,很容易发现,s[i] 和 s[i-1]的关系可以影响到01串能否消掉如果 s[i] == s[i-1] 则这两个为 1 1 或者是 0 0 ,这两个都是不可以消掉的,只有s[i]!=s[i-1]的时候才可以抵消为一个字符例如 0000000001最后一个1一直可以通过变换往前移动
01 -> 1 所以他可以组成 n-1个子串 01, 001,0001,00001,000001,0000001等等.
AC代码:
#include<bits/stdc++.h>
using namespace std;
#define ll long long
int t,n;
string s;
void solve(){
cin>>n;
cin>>s;
s=" "+s;
ll ans=n;
s[0]=s[1];
for(int i=n;i>=1;i--){
if(s[i]!=s[i-1]){
ans+=(ll)(i-1);
}
}
cout<<ans<<endl;
}
int main(){
ios::sync_with_stdio(0);
cin.tie(0),cout.tie(0);
cin>>t;
while(t--){
solve();
}
return 0;
}