lintcode 843. 数字翻转 个人想法及C++代码

今天做lintcode算法题的时候，一开始被这个题难住了，第一想法就是去搜答案，但是网上没有找到，没想到最后居然能自己给做出来，看到最后通过的时候开心的无以复加，可能这就是刷题最大的乐趣吧。这里把代码和思路贴出来给大家分享一下，希望可以帮助到有需要的人，如果有不对的和可以改进的，欢迎大家批评指正。

原题：

给你一个01构成的数组。请你找出最小翻转步数，使得数组满足以下规则：
1的后面可以是1或者0，但是0的后面必须是0。

 注意事项

• 输入的数组长度n <= 100000。

样例

给出 array = [1,0,0,1,1,1] , 返回2。

解释：
把两个0翻转成1。

给出 array = [1,0,1,0,1,0] , 返回2。

解释：
把第二个1和第三个1都翻转成0。

这是一个动态规划题，主要的思路是根据数组翻转前一步最后一位是0还是1来计算这一步翻成结尾是0或1各需要翻转几次。

用二维的容器保存状态，dp[i][j]，到数组第i位，以j结尾时需要翻转几步，j的取值只有0或者1。

class Solution {
public:
    /**
     * @param nums: the array
     * @return: the minimum times to flip digit
     */
    int flipDigit(vector<int> &nums) {
        // Write your code here
        int size=nums.size();
        vector<vector<int> > dp(size+1,vector<int>(2,0));
        
        for(int i=1;i<=size;++i)
        {
            if(nums[i-1])
            {
                dp[i][0]=min(dp[i-1][0]+1,dp[i-1][1]+1);
                dp[i][1]=dp[i-1][1];
            }
            else
            {
                dp[i][0]=min(dp[i-1][0],dp[i-1][1]);
                dp[i][1]=dp[i-1][1]+1;
            }
        }
        
        return dp[size][0]<dp[size][1]?dp[size][0]:dp[size][1];
    }
};

附上代码，新手上路，欢迎批评指正，或者在评论区大家一起讨论！