题目来自力扣 1394. 找出数组中的幸运数 - 力扣(LeetCode)
记录每日刷题学习过程,如有侵权问题,联系作者删除。
欢迎有意向的同学们交流学习。
一、题目描述
在整数数组中,如果一个整数的出现频次和它的数值大小相等,我们就称这个整数为「幸运数」。
给你一个整数数组 arr
,请你从中找出并返回一个幸运数。
- 如果数组中存在多个幸运数,只需返回 最大 的那个。
- 如果数组中不含幸运数,则返回 -1 。
示例1:
输入:arr = [2,2,3,4]
输出:2
解释:数组中唯一的幸运数是 2 ,因为数值 2 的出现频次也是 2 。
示例2:
输入:arr = [1,2,2,3,3,3]
输出:3
解释:1、2 以及 3 都是幸运数,只需要返回其中最大的 3 。
示例3:
输入:arr = [2,2,2,3,3]
输出:-1
解释:数组中不存在幸运数。
二、思路笔记
1、幸运数:在数组中,出现频次等于自身大小的数字
2、只返回最大的幸运数,若不存在幸运数,返回 -1
遍历求得每个数字的频次存入字典,再在其中找出最大的幸运数
三、代码
class Solution:
def findLucky(self, arr: List[int]) -> int:
d = {}
for i in arr:
if i not in d:
d[i] = 1
else:
d[i] += 1
lucky_num = 0
arr_set = list(set(arr))
for k in arr_set:
if k == d[k]:
if k > lucky_num:
lucky_num = k
elif k == arr_set[-1] and k != d[k] and lucky_num == 0:
lucky_num = -1
return lucky_num