题意
给你一串01串,其中有一部分字符为?号表示这个字符串既可以被0又可以被1替换,让你求出“交替01串”的个数(00和11和001之类有连续两个字母相同的字符串就不能算“交替01串”)
举例:”0?10“ 这个字符串
有8种情况:0,0?,?,?1,?10,1,10,0满足题意
题解
初始化:
对于当前位置为0的位置,将dp[i][0]
初始化为1,dp[i][1]
初始化为0
对于当前位置为1的位置,将dp[i][0]
初始化为0,dp[i][1]
初始化为1
对于当前位置为”?“的位置,将dp[i][0]
初始化为1,dp[i][1]
初始化为1
思路
先进行特判,当n=1
时,直接输出1即可
对于n>=2
的情况,dp[i][j]
数组表示当前是第i位并且取j的时候的贡献。
当j=0
时,它的贡献是前一个字符为1的贡献+1
dp[i][0]=dp[i-1][1]+1;
dp[i][1]=0;
当j=1
时,它的贡献是前一个字符为0的贡献+1
dp[i][1]=dp[i-1][0]+1;
dp[i][0]=0;
当j为“?"时,他的贡献是前一个字符为0的贡献+1和前一个字符为1的贡献+1
dp[i][0]=dp[i-1][1]+1;
dp[i][1]=dp[i-1][0]+1;
答案从头到尾扫一遍加上max(dp[i][0],dp[i][1])
即可
代码
#include <bits/stdc++.h>
using namespace std;
#define ll long long
int main()
{
ll t,n;
cin>>t;
ll dp[200007][2],k[200009];
while(t--)
{
ll ans=0;
string s;
cin>>s;
n=s.size();
if(n==1)
{
cout<<"1"<<endl;
continue;
}
if(s[0]=='0')
{
dp[0][0]=1;
dp[0][1]=0;
}
else if(s[0]=='1')
{
dp[0][1]=1;
dp[0][0]=0;
}
else
{
dp[0][1]=1;
dp[0][0]=1;
}
for(int i=1; i<=n-1; i++)
{
if(s[i]=='0')
{
dp[i][0]=dp[i-1][1]+1;
dp[i][1]=0;
}
else if(s[i]=='1')
{
dp[i][1]=dp[i-1][0]+1;
dp[i][0]=0;
}
else
{
dp[i][0]=dp[i-1][1]+1;
dp[i][1]=dp[i-1][0]+1;
}
}
for(int i=0;i<=n-1;i++)
ans=ans+max(dp[i][0],dp[i][1]);
cout<<ans<<endl;
}
}
一点吐槽
考试的时候这道题弹出来了4个提示。。。
一开始我们连样例都没有看懂(笑)
wa了一发是因为没开longlong。。
以后这个方面还是得注意一下。