今天做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];
}
};
附上代码,新手上路,欢迎批评指正,或者在评论区大家一起讨论!