程序在Python中对颜色列表进行分区

假设我们有一个颜色字符串列表，其中可能包含“红色”，“绿色”和“蓝色”，我们必须对列表进行分区，以使红色位于绿色之前，绿色位于蓝色之前。

因此，如果输入的颜色像[color] = [“ blue”，“ green”，“ blue”，“ red”，“ red”]，那么输出将是['red'，'red'，'green'，'蓝色”，“蓝色”]

为了解决这个问题，我们将遵循以下步骤-

• 绿色：= 0，蓝色：= 0，红色：= 0

• 对于strs中的每个字符串，执行

• strs [blue]：=“蓝色”

• 蓝色：=蓝色+ 1

• strs [blue]：=“蓝色”

• 蓝色：=蓝色+ 1

• strs [green]：=“绿色”

• 绿色：=绿色+ 1

• strs [blue]：=“蓝色”

• 蓝色：=蓝色+ 1

• strs [green]：=“绿色”

• 绿色：=绿色+ 1

• strs [red]：=“红色”

• 红色：=红色+ 1

• 如果字符串与“红色”相同，则

• 否则，当字符串与“绿色”相同时，则

• 否则，当字符串与“ blue”相同时，则

• 返回str

让我们看下面的实现以更好地理解-

示例

class Solution:
   def solve(self, strs):
      green = 0
      blue = 0
      red = 0
      for string in strs:
         if string == "red":
            strs[blue] = "blue"
            blue += 1
            strs[green] = "green"
            green += 1
            strs[red] = "red"
            red += 1
         elif string == "green":
            strs[blue] = "blue"
            blue += 1
            strs[green] = "green"
            green += 1
         elif string == "blue":
            strs[blue] = "blue"
            blue += 1
      return strs
ob = Solution()colors = ["blue","green", "blue", "red", "red"]
print(ob.solve(colors))

输入项

["blue","green", "blue", "red", "red"]

输出结果

['red', 'red', 'green', 'blue', 'blue']