leetcode

75. Sort Colors.go

2019-03-23  本文已影响0人  AnakinSun

简单暴力

func swap(nums []int, i, j int) {
   tmp := nums[i]
   nums[i] = nums[j]
   nums[j] = tmp

func sortColors(nums []int) {
   l, i, r := 0, 0, len(nums)-1
   for i <= r {
       if nums[i] == 0 {
           swap(nums, i, l)
           i++
           l++
       } else {
           if nums[i] == 1 {
               i++
           } else {
               swap(nums, i, r)
               r--
           }
       }
   }
}
上一篇下一篇

猜你喜欢

热点阅读