[Swift]LeetCode324. 摆动排序 II | Wiggle Sort II
发布时间:2020-12-14 05:06:16 所属栏目:百科 来源:网络整理
导读:Given an unsorted array? nums ,reorder it such that? nums[0] nums[1] nums[2] nums[3]... . Example 1: Input: Output: One possible answer is .nums = [1,5,1,6,4][1,4,6] Example 2: Input: Output: One possible answer is .nums = [1,3,2,1][2,2] N
Given an unsorted array? Example 1: Input: Output: One possible answer is .nums = [1,5,1,6,4][1,4,6] Example 2: Input: Output: One possible answer is .nums = [1,3,2,1][2,2] Note: Follow Up: 给定一个无序的数组? 示例?1: 输入: 输出: 一个可能的答案是 nums = [1,6] 示例 2: 输入: 输出: 一个可能的答案是 nums = [1,2] 说明: 进阶: 380ms 1 class Solution { 2 func wiggleSort(_ nums: inout [Int]) { 3 4 if nums.count < 2 { 5 return 6 } 7 8 var count = nums.count 9 10 var begin = 0,end = count - 1,mid = (begin + end) >> 1 11 12 func patition(_ begin : Int,_ end : Int) -> Int { 13 let index = nums[begin] 14 var low = begin,high = end 15 while low < high { 16 while low < high && nums[high] >= index { 17 high -= 1 18 } 19 if low < high { 20 nums[low] = nums[high] 21 low += 1 22 } 23 24 while low < high && nums[low] <= index { 25 low += 1 26 } 27 28 if low < high { 29 nums[high] = nums[low] 30 high -= 1 31 } 32 33 } 34 35 nums[low] = index 36 37 return low 38 } 39 40 var index = patition(begin,end) 41 42 while index != mid { 43 if index < mid { 44 begin = index + 1 45 }else { 46 end = index - 1 47 } 48 49 index = patition(begin,end) 50 } 51 52 mid = nums[index] 53 54 var n = count 55 56 func A(_ i : Int) -> Int { 57 return (1+2*i)%(count|1) 58 } 59 var i = 0,j = 0,k = count - 1 60 while j <= k { 61 if nums[A(j)] > mid { 62 nums.swapAt(A(i),A(j)) 63 i+=1 64 j+=1 65 }else if nums[A(j)] < mid { 66 nums.swapAt(A(j),A(k)) 67 k-=1 68 }else { 69 j+=1 70 } 71 } 72 } 73 } (编辑:李大同) 【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容! |