题目大意: t t t组询问,每次给定一个仅包含字符 1 1 1或 0 0 0或 ? ? ?字符串 s s s,定义一个子串是不稳定的当且仅当子串中任意相邻两数均不相同,如 101010... 101010... 101010...或 010101... 010101... 010101...其中 ? ? ?可以变为 1 1 1或 0 0 0其中一种。请求出给定的 s s s中最多可以有的不稳定子串个数
一道简单的
d
p
dp
dp题,一开始理解错题意,还以为样例错了(
题意的意思是先选出子串,再判断,例如样例
0
?
10
0?10
0?10,符合条件的是:
0
0
0,
1
1
1,
?
?
?,
0
?
0?
0?,
?
1
?1
?1,
10
10
10,
?
10
?10
?10,其中单个
?
?
?算两种,因此答案为8
显然我们可以用
d
p
[
N
]
[
2
]
dp[N][2]
dp[N][2]来表示串
s
s
s在第
i
i
i位时为不稳定子串的个数
初始化
d
p
dp
dp:
i
f
:
s
[
i
]
=
=
′
1
′
,
d
p
[
i
]
[
1
]
=
1
if:s[i]=='1',dp[i][1]=1
if:s[i]==′1′,dp[i][1]=1
i
f
:
s
[
i
]
=
=
′
0
′
,
d
p
[
i
]
[
0
]
=
1
if:s[i]=='0',dp[i][0]=1
if:s[i]==′0′,dp[i][0]=1
i
f
:
s
[
i
]
=
=
′
?
′
,
d
p
[
i
]
[
0
]
=
d
p
[
i
]
[
1
]
=
1
if:s[i]=='?',dp[i][0]=dp[i][1]=1
if:s[i]==′?′,dp[i][0]=dp[i][1]=1
显然易见可得转移方程为:
i
f
:
s
[
i
]
=
=
′
1
′
,
d
p
[
i
]
[
1
]
=
d
p
[
i
−
1
]
[
0
]
+
1
if:s[i]=='1',dp[i][1]=dp[i-1][0]+1
if:s[i]==′1′,dp[i][1]=dp[i−1][0]+1
i
f
:
s
[
i
]
=
=
′
0
′
,
d
p
[
i
]
[
0
]
=
d
p
[
i
−
1
]
[
1
]
+
1
if:s[i]=='0',dp[i][0]=dp[i-1][1]+1
if:s[i]==′0′,dp[i][0]=dp[i−1][1]+1
i
f
:
s
[
i
]
=
=
′
?
′
,
d
p
[
i
]
[
0
]
=
d
p
[
i
−
1
]
[
1
]
+
1
,
d
p
[
i
]
[
1
]
=
d
p
[
i
−
1
]
[
0
]
+
1
if:s[i]=='?',dp[i][0]=dp[i-1][1]+1,dp[i][1]=dp[i-1][0]+1
if:s[i]==′?′,dp[i][0]=dp[i−1][1]+1,dp[i][1]=dp[i−1][0]+1
最后一步求
a
n
s
ans
ans
a
n
s
+
=
∑
i
=
0
l
e
n
−
1
m
a
x
(
d
p
[
i
]
[
0
]
,
d
p
[
i
]
[
1
]
)
ans+=\sum_{i=0}^{len-1}max(dp[i][0],dp[i][1])
ans+=∑i=0len−1max(dp[i][0],dp[i][1])
该听离散课了,逃
AC代码:
#include <bits/stdc++.h>
using namespace std;
const int N=2e5+10;
typedef long long ll;
int t;
string s;
ll dp[N][2];
int main(){
cin>>t;
while(t--){
memset(dp,0,sizeof(dp));
cin>>s;
for(int i=0;i<s.size();i++) {
if(s[i]=='0') dp[i][0]=1;
if(s[i]=='1') dp[i][1]=1;
if(s[i]=='?') dp[i][0]=1,dp[i][1]=1;
}
for(int i=1;i<s.size();i++){
if(s[i]=='0') dp[i][0]=dp[i-1][1]+1;
if(s[i]=='1') dp[i][1]=dp[i-1][0]+1;
if(s[i]=='?') dp[i][0]=dp[i-1][1]+1,dp[i][1]=dp[i-1][0]+1;
}
ll max1=0,res=0;
for(int i=0;i<s.size();i++){
max1=max(dp[i][1],dp[i][0]);
res+=max1;
}
cout<<res<<endl;
}
return 0;
}