程序通过向右旋转 k 次来找到第 i 个元素
假设我们有一个数组nums,一个值k和另一个值i。我们必须在将nums的元素向右旋转k次后找到索引i处的元素。
因此,如果输入类似于nums=[2,7,9,8,10]k=3i=2,则输出将为10,因为在第三次旋转后数组将为[9,8,10,2,7],所以现在第i个元素将是nums[2]=10。
示例
让我们看下面的实现来更好地理解
def solve(nums, k, i): for r in range(k): nums.insert(0, nums.pop()) return nums[i] nums = [2,7,9,8,10] k = 3 i = 2 print(solve(nums, k, i))
输入
[2,7,9,8,10] , 3, 2输出结果
10