LeetCode破解之收集

102 阅读2分钟

持续创作,加速成长!这是我参与「掘金日新计划 · 6 月更文挑战」的第4天,点击查看活动详情

题目描述

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

示例 1:

输入:leaves = "rrryyyrryyyrr"

输出:2

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

动态规划

这个题目我的初步思路是:我一开始想到的是动态规划,但是卡住了,然后发现可以分段,于是我们可以先把要求的字符串分为3段,第i个字母可能处于0,1,2三段,dp数组的第一个维度为3。然后我们可以构建一个二维数组,利用三个状态来表示三个不同的区域颜色集合,循环遍历字符串的每一个字符,第一个字符只能是状态0,也就是只能是r字符,第二个字符可以是状态0,也可以是状态1。第三个字符以后的字符,每一个字符都会有三种状态。当为状态0时,前面的所有字符都是状态0;当为状态1时,当前字符的前一个字符可以是状态0,也可以是状态1,所以通过比较大小,选取小的那个状态;当为状态2时,当前字符的前一个字符可以是状态1,也可以是状态2,但是不可以是状态0,因为要保证每一个字符至少有一个,也是选取小的那个状态。最后选取result[leaves.length() - 1][2]来作为最终的结果。

class Solution {
    public int minimumOperations(String leaves) {
        int n = leaves.length();
        int[][] dp = new int[3][n];

        if(leaves.charAt(0) == 'y')    dp[0][0] = 1;
        for(int i = 1; i < n; i++)
        {
            dp[0][i] = dp[0][i - 1];
            if(leaves.charAt(i) == 'y')    dp[0][i]++;
        }

        dp[1][0] = dp[0][0];
        for(int i = 1; i < n; i++)
        {
            dp[1][i] = dp[1][i - 1];
            int temp = dp[0][i - 1];
            if(leaves.charAt(i) == 'r')
            {
                dp[1][i]++;
                temp++;
            }
            dp[1][i] = Math.min(dp[1][i], temp);
        }

        dp[2][0] = dp[1][0];
        dp[2][1] = dp[1][1];
        for(int i = 2; i < n; i++)
        {
            dp[2][i] = dp[2][i - 1];
            int temp = dp[1][i - 1];
            if(leaves.charAt(i) == 'y')
            {
                dp[2][i]++;
                temp++;
            }
            dp[2][i] = Math.min(dp[2][i], temp);
        }

        return dp[2][n - 1];
    }
}

最后

题目没读懂一开始,后面才发现可以用大量备用树叶去替换数组中的r和y,而不是数组内部进行r和y的位置交换。