秋叶收藏集

题目描述:

        小扣出去秋游,途中收集了一些红叶和黄叶,他利用这些叶子初步整理了一份秋叶收藏集 leaves, 字符串 leaves 仅包含小写字符 r 和 y, 其中字符 r 表示一片红叶,字符 y 表示一片黄叶。 出于美观整齐的考虑,小扣想要将收藏集中树叶的排列调整成「红、黄、红」三部分。每部分树叶数量可以不相等,但均需大于等于 1。每次调整操作,小扣可以将一片红叶替换成黄叶或者将一片黄叶替换成红叶。请问小扣最少需要多少次调整操作才能将秋叶收藏集调整完毕。

示例:

        输入:leaves = "rrryyyrryyyrr"

        输出:2

        解释:调整两次,将中间的两片红叶替换成黄叶,得到 "rrryyyyyyyyrr"

题解:

        这道题其实很符合动态规划的特征,即后续调整需要根据前序状态来决定,但是不同的是每个位置的树叶其实是有三个状态的:1.左边的红色 2.中间的黄色 3.右边的红色。不妨建立二维数组dp[i][j],其中第一维的i代表了到第i个叶子需要的最小的调整次数,第二维j代表了i树叶的状态。对于状态1来说即dp[i][0](0代表状态1,后续以此类推),dp[i][0] = dp[i-1][0] + isyellow;即如果第i个叶子置为状态1那么其前面叶子的状态也只能为1,同时判断需不需要将i叶子本身进行换色。dp[i][1] = min(dp[i-1][0], dp[i-1][1]) + isred。对于第i个叶子置为状态2的情况,它的前序既可以是状态1,也可以是状态2,同时考虑是否本身需要换色。同理,dp[i][2] = min(dp[i-1][1], dp[i-1][2]) + isyellow。对于第i个叶子是状态3的情况,它的前序叶子状态只能是2或3,同时 同时考虑是否本身需要换色。最后,由于状态的数量不能大于叶子的数量,所以dp[0][1] = dp[0][2] = dp[1][2] = INT_MAX;边界条件为:第0个叶子必须为红色。

代码如下:

class Solution {
public:
    int minimumOperations(string leaves) {
        if(leaves.empty()){
            return 0;
        }
        int n = leaves.length();
        vector<vector<int>> dp(n,vector<int>(3));
        dp[0][0] = leaves[0] == 'r' ? 0 : 1;
        dp[0][1] = dp[0][2] = dp[1][2] = INT_MAX;
        for(int i = 1; i < n; i++){
            cout<<i<<endl;
            int isyellow = static_cast<int>(leaves[i] == 'y');
            int isred = static_cast<int>(leaves[i] == 'r');
            dp[i][0] = dp[i-1][0] + isyellow;
            dp[i][1] = min(dp[i-1][0], dp[i-1][1]) + isred;
            if(i >= 2){
                dp[i][2] = min(dp[i-1][1], dp[i-1][2]) + isyellow;
            }
        }
        return dp[n-1][2];
    }
};

  • 12
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值