CF1535C Unstable String

CF传送门

题目大意：$t$组询问，每次给定一个仅包含字符$1$或$0$或$?$字符串$s$,定义一个子串是不稳定的当且仅当子串中任意相邻两数均不相同，如$101010...$或$010101...$其中$?$可以变为$1$或$0$其中一种。请求出给定的$s$中最多可以有的不稳定子串个数

一道简单的$dp$题，一开始理解错题意，还以为样例错了（
题意的意思是先选出子串，再判断，例如样例$0?10$,符合条件的是:$0$,$1$,$?$,$0?$,$?1$,$10$,$?10$，其中单个$?$算两种，因此答案为8

显然我们可以用$dp[N][2]$来表示串$s$在第$i$位时为不稳定子串的个数
初始化$dp$：
$if:s[i]=='1',dp[i][1]=1$
$if:s[i]=='0',dp[i][0]=1$
$if:s[i]=='?',dp[i][0]=dp[i][1]=1$
显然易见可得转移方程为：
$if:s[i]=='1',dp[i][1]=dp[i-1][0]+1$
$if:s[i]=='0',dp[i][0]=dp[i-1][1]+1$
$if:s[i]=='?',dp[i][0]=dp[i-1][1]+1,dp[i][1]=dp[i-1][0]+1$
最后一步求$ans$
$ans+=\sum_{i=0}^{len-1}max(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;
}