IOS 算法(中级篇) ----- 构造元素不等于两相邻元素平均

给你一个 下标从 0 开始 的数组 nums ，数组由若干 互不相同的 整数组成。你打算重新排列数组中的元素以满足：重排后，数组中的每个元素都 不等于 其两侧相邻元素的 平均值 。更公式化的说法是，重新排列的数组应当满足这一属性：对于范围 1 <= i < nums.length - 1 中的每个 i ，(nums[i-1] + nums[i+1]) / 2 不等于 nums[i] 均成立 。
返回满足题意的任一重排结果。
3 <= nums.length <= 105
0 <= nums[i] <= 105

例子

输入： nums = [1,2,3,4,5]
输出： [1,2,4,5,3]
解释：
i=1, nums[i] = 2, 两相邻元素平均值为 (1+4) / 2 = 2.5
i=2, nums[i] = 4, 两相邻元素平均值为 (2+5) / 2 = 3.5
i=3, nums[i] = 5, 两相邻元素平均值为 (4+3) / 2 = 3.5

输入：nums = [6,2,0,9,7]
输出：[9,7,6,2,0]
解释：
i=1, nums[i] = 7, 两相邻元素平均值为 (9+6) / 2 = 7.5
i=2, nums[i] = 6, 两相邻元素平均值为 (7+2) / 2 = 4.5
i=3, nums[i] = 2, 两相邻元素平均值为 (6+0) / 2 = 3

解题思路

滑动窗口(双指针)

其实这道题不难, 思路找对的话很容易处理
我这边是将数组变化为 [小, 大, 小, 大, 小, 大 ......] 这样形式返回

左右2边都比中间小, 那么平均值比中间小
左右2边都比中间大, 那么平均值比中间大
满足题目要求 (nums[i-1] + nums[i+1]) / 2 不等于 nums[I]

1.将数组正序排序
2.设置2个指针, 依次改变
3.指针+2, 从index=1开始以长度为2窗口滑动, 依次把窗口内2个位置互换

例子1 例子2 例子3

代码

未翻译版

    func rearrangeArray(_ nums: [Int]) -> [Int] {
        
        var res = nums.sorted(), last = 1, next = 2
        
        while next < nums.count {
            
            let temp = res[last]
            res[last] = res[next]
            res[next] = temp
            
            last += 2
            next += 2
        }
        
        return res
    }

翻译版

    func rearrangeArray(_ nums: [Int]) -> [Int] {
        
        // 数组正序排序, 并设置2个指针
        var res = nums.sorted(), last = 1, next = 2
        
        // 循环, 滑动窗口
        while next < nums.count {
            
            // 互换位置 
            let temp = res[last]
            res[last] = res[next]
            res[next] = temp
            
            // 指针+2, 滑动窗口
            last += 2
            next += 2
        }
        
        // 返回结果
        return res
    }

题目来源：力扣（LeetCode） 感谢力扣爸爸 :)
IOS 算法合集地址