题目:
本题是经典的「荷兰国旗问题」,由计算机科学家 Edsger W. Dijkstra 首先提出。
给定一个包含红色、白色和n个元素的数组nums,原地对它们进行排序,使得相同颜色的元素相邻,并按照红色、白色、蓝色顺序排列。我们使用整数0、1和2分别表示红色、白色和蓝色。必须在不使用库内置的sort函数的情况下解决这个问题。
示例 :
输入 | 输出 | 解释 |
---|---|---|
nums = [2,0,2,1,1,0] | [0,0,1,1,2,2] |
提示:
- 0==nums.length
- 1<=n<=300
- nums[i]为0、1或2
相关标签:数组、排序、双指针
题解:
根据题目中的提示,我们可以统计出数组中 0,1,2 的个数,再根据它们的数量,重写整个数组。
代码:Python
import collections
nums = [2, 0, 2, 1, 1, 0]
def leetcode():
dict_count, n = collections.Counter(nums), 0
for i in range(3):
for j in range(dict_count[i]):
nums[n] = i
n += 1
return nums
if __name__ == '__main__':
res = leetcode()
print(res) # [0, 0, 1, 1, 2, 2]
总结:
题目内容来源于leetcode链接:颜色分类