2022-01-07 每日打卡:难题精刷
写在前面
“这些事儿在熟练之后,也许就像喝口水一样平淡,但却能给初学者带来巨大的快乐,我一直觉得,能否始终保持如初学者般的热情、专注,决定了在做某件事时能走多远,能做多好。” 该系列文章由python编写,所刷题目共三个来源:之前没做出来的 ;Leetcode中等,困难难度题目; 周赛题目;某个专题的经典题目,所有代码已AC。每日1-3道,随缘剖析,希望风雨无阻,作为勉励自己坚持刷题的记录。
2122. 还原原数组
这个题分析一下可以知道,让我们寻找间隔相同的两个数组。但是间隔是无法通过数学方法计算出来的,使用枚举。枚举的开始是最小的元素or最大的元素,一定是属于较小or较大数组的。
在遍历过程中注意一点,需要在含有相同元素的数字的列表中标记,无法使用搜索,而只能使用遍历。另外,像是while(i<n){i++}
这样的循环体,考察边界,如果满足条件还是会加一的。
class Solution:
def recoverArray(self, nums: List[int]) -> List[int]:
nums.sort()
n, lenth, ans =len(nums), len(nums)//2, []
for _ in range(1,len(nums)):
# 枚举每个k的可能性
k = (nums[_]-nums[0])
if k ==0 or k%2!=0:
continue
l, used, ans, k = 0, [False for _ in range(n)],[], k//2
for r in range(1,n):
while r<n-1 and (used[r] or nums[r]!=nums[l]+2*k):
r+=1
used[l] = used[r] = True
ans.append(nums[l]+k)
if r == n-1:
break
while l<n-1 and used[l]:
l += 1
r = l
if len(ans) == lenth:
return ans
2121. 相同元素的间隔之和
很类似之前做的接水问题,我们把需要重复计算的部分提取出来,而不是通过循环计算。比如上题3的位置,对于第一个3而言,其下标为2,则遍历第二个3时,加入5-2=3,遍历到第三个3时,加入3+6-5=4即可,而此时也同时加入第二个3的位置为6-5。
对于每个有重复数字的位置而言,其最终结果=左侧数字的贡献+右侧数字的贡献
,通过两次遍历分别正向和逆向统计出结果即可。
这里的res是针对每个下标单独计算,而total和cnt可以重复利用,不断累加。
class Solution:
def getDistances(self, arr: List[int]) -> List[int]:
n = len(arr)
res = [0] * n
# 每个数值出现下标之和
total = defaultdict(int)
# 每个数值出现次数
cnt = defaultdict(int)
# 正向遍历并更新两个哈希表以及间隔之和数组
for i in range(n):
val = arr[i]
if val in cnt:
res[i] += i * cnt[val] - total[val]
total[val] += i
cnt[val] += 1
# 清空哈希表
total.clear()
cnt.clear()
# 反向遍历并更新两个哈希表以及间隔之和数组
for i in range(n - 1, -1, -1):
val = arr[i]
if val in cnt:
res[i] += total[val] - i * cnt[val]
total[val] += i
cnt[val] += 1
return res