268. 丢失的数字
- 题目链接
- 解题思路:利用原有的数组,i位置上的数就是
nums[i],但是这里有一个问题,就是如果这个数是n,怎么办?n不能够放到nums[n]上。用一个变量n_index来存n这个数的位置(假设有n这个数),初始化为-1
-
- 第一种情况:没有n这个数,那么
[0, n-1]上都已经放好了,此时n_index为-1 - 第二种情况:有n这个数,那么
n_index的位置,就是缺少的那个数
- 第一种情况:没有n这个数,那么
-
代码
class Solution: def missingNumber(self, nums: List[int]) -> int: n = len(nums) n_index = -1 i = 0 while i < n: # [0, i - 1]都满足 i位置上的数放在nums[i]上,即nums[i] == i if nums[i] == i: i += 1 elif nums[i] == n: n_index = i i += 1 else: # 把i这个数放到nums[i]上 tmp = nums[i] nums[i] = nums[tmp] nums[tmp] = tmp return n if n_index == -1 else n_index

浙公网安备 33010602011771号