Python算法指南:程序员经典算法分析与实现
上QQ阅读APP看书,第一时间看更新

例70 数组划分

1. 问题描述

给出一个整数数组nums和一个整数k。划分数组(即移动数组nums中的元素),使得:①所有小于k的元素移到左边;②所有大于等于k的元素移到右边。返回数组划分的位置,即数组中第一个位置i,满足nums[i]≥k

2. 问题示例

给出数组nums=[3,2,2,1]和k=2,返回1。

3. 代码实现

4. 运行结果

输入:[5,1,4,2,3] 2

输出:1