程序通过向右旋转 k 次来找到第 i 个元素

假设我们有一个数组 nums,一个值 k 和另一个值 i。我们必须在将 nums 的元素向右旋转 k 次后找到索引 i 处的元素。

因此,如果输入类似于 nums = [2,7,9,8,10] k = 3 i = 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

以上是 程序通过向右旋转 k 次来找到第 i 个元素 的全部内容, 来源链接: utcz.com/z/345678.html

回到顶部