TopCoder SRM 650 Div2 Problem 500 - TaroFillingAStringDiv2 (DP)

题意

有一串A、B、?组成的字符串,现在要求输出让该字符串连续相同的字符数最少的数量。

思路

dp。

$dp(i, j)表示前i个字符,当前位置是j字符的最少的数量$

\[dp(i, j) = \begin{cases} min(dp(i-1, 0)+1, dp(i-1, 1)) & \text{j == 0} \\\ min(dp(i-1, 0), dp(i-1, 1)+1) & \text{j==1} \\\ \end{cases} \\\ \text{if(str[i] == '?')} \\\ dp(i, 0) = min(dp(i-1, 0) + 1, dp(i-1, 1)) \\\ dp(i, 1) = min(dp(i-1, 0), dp(i-1, 1) + 1)\]

代码

class TaroFillingAStringDiv2 {
    public:
    int getNumber(string str) {
        if (str.empty() || SZ(str) == 1) return 0;
        MS(dp, INF);
        dp[0][0] = dp[0][1] = 0;
        if (str[0] == 'A') dp[0][1] = INF;
        if (str[0] == 'B') dp[0][0] = INF;
        for (int i = 1; i < SZ(str); i++)
        {
            if (str[i] != '?')
            {
                if (str[i] == 'A')
                {
                    dp[i][0] = min(dp[i-1][1], dp[i-1][0] + 1);
                    dp[i][1] = INF;
                }
                else
                {
                    dp[i][1] = min(dp[i-1][1] + 1, dp[i-1][0]);
                    dp[i][0] = INF;
                }
            }
            else
            {
                dp[i][0] = min(dp[i-1][1], dp[i-1][0] + 1);
                dp[i][1] = min(dp[i-1][1] + 1, dp[i-1][0]);
            }
        }
        int len = SZ(str) - 1;
        return min(dp[len][0], dp[len][1]);
    }
};

Powered by Jekyll and Theme by solid