红绿正方形染色问题

原文地址:

题目描述

有一些排成一行的正方形。每个正方形已经被染成红色或者绿色。现在可以选择任意一个正方形然后用这两种颜色的任意一种进行染色,这个正方形的颜色将会被覆盖。目标是在完成染色之后,每个红色 R 都比每个绿色 G 距离最左侧近。 返回最少需要涂染几个正方形,示例: s = RGRGR,我们涂染之后变成 RRRGG 满足要求了,涂染的个数为 2, 没有比这个更好的涂染方案。

要满足染色后,每个红色 R 都比每个绿色 G 距离最左侧近,只有可能是下述三种情况:

定义两个数组

int[] leftG = new int[N];

leftG[i]表示 i 位置的左边包括 i 在内有几个 G。

通过从右往左遍历一次数组可以预处理得到 leftG数组。

int[] rightR = new int[N];

rightR[i]表示 i 位置右边包括 i 在内有几个 R。

通过从左往右遍历一次数组可以预处理得到 rightR数组

如果 rightR[0] == N || leftG[N - 1] == N,说明数组全为 R 或者全为 G,此时无需染色,直接返回 0。

如果是普遍情况,那就直接判断每个位置为分割点,左侧都变为 R,右侧都变为 G,需要的最小染色次数是多少,即: leftG[i] + rightR[i] - 1,之所以要减 1 是因为重复算了 i 位置的情况。

完整代码见

import java.util.Scanner;
// R都在左边,G都在右边,或者全G,全R
public class Main {

    // 两个预处理数组
    // TODO 空间方面可以优化
    public static int minColors(String str) {
        if (str == null || str.length() = 0; i--) {
            if (strs[i] == 'R') {
                if (i == N - 1) {
                    rightR[i]++;
                } else {
                    rightR[i] = rightR[i + 1] + 1;
                }
            } else {
                if (i != N - 1) {
                    rightR[i] = rightR[i + 1];
                }
            }
        }

        // 全R或者全G的情况
        if (rightR[0] == N || leftG[N - 1] == N ) {
            return 0;
        }

        int min = N;
        for (int i = 0; i < N; i++) {
            // 之所以要-1是因为重复算了i位置的处理情况
            min = Math.min(leftG[i] + rightR[i] - 1, min);
        }
        return min;
    }

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        String colors = in.nextLine();
        System.out.println(minColors(colors));
        in.close();
    }
}

空间复杂度 O(N),时间复杂度 O(N)

Original: https://www.cnblogs.com/greyzeng/p/16728462.html
Author: Grey Zeng
Title: 红绿正方形染色问题

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/604881/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球