Leetcode 【2342. 数位和相等数对的最大和】

给你一个下标从 0 开始的数组 nums ,数组中的元素都是 整数。请你选出两个下标 iji != j),且 nums[i] 的数位和 与 nums[j] 的数位和相等。

请你找出所有满足条件的下标 ij ,找出并返回nums[i] + nums[j]可以得到的 最大值

示例 1:

复制代码
输入:nums = [18,43,36,13,7]
输出:54
解释:满足条件的数对 (i, j) 为:
- (0, 2) ,两个数字的数位和都是 9 ,相加得到 18 + 36 = 54 。
- (1, 4) ,两个数字的数位和都是 7 ,相加得到 43 + 7 = 50 。
所以可以获得的最大和是 54 。

示例 2:

复制代码
输入:nums = [10,12,19,14]
输出:-1
解释:不存在满足条件的数对,返回 -1 。

提示:

  • 1 <= nums.length <=
  • 1 <= nums[i] <=

方法:哈希表

看到这么恐怖的数量级,第一想法肯定是暴力行不通,那么就来试试哈希表

首先将数组nums中每个元素的数位和保存在一个新数组中:

python 复制代码
for i in nums:
            str_i = str(i)
            res = 0
            for digit in str_i:
                res += int(digit)
            sum_num.append(res)

这段代码的时间复杂度是O(n)

然后创建字典,key值为当前数位和,当字典中不存在该key值时,dic[key]为nums中的当前元素值,如果字典中存在key值,那么将新的nums元素值append到字典中:

python 复制代码
for i in range(n):
            key = sum_num[i]
            if key not in dic:
                dic[key] = [nums[i]]
            else:
                dic[key].append(nums[i])

最后遍历字典,找出key值对应的value有两个及以上,并且按照降序排列,找出最大和即可:

python 复制代码
for key, numbers in dic.items():
            if len(numbers) >= 2:
                numbers.sort(reverse=True)
                pair_sum = numbers[0] + numbers[1]
                ans = max(ans, pair_sum)

完整代码:

python 复制代码
class Solution:
    def maximumSum(self, nums: List[int]) -> int:
        sum_num = []
        dic={}
        n=len(nums)
        ans=-1
        for i in nums:
            str_i = str(i)
            res = 0
            for digit in str_i:
                res += int(digit)
            sum_num.append(res)

        for i in range(n):
            key = sum_num[i]
            if key not in dic:
                dic[key] = [nums[i]]
            else:
                dic[key].append(nums[i])

        for key, numbers in dic.items():
            if len(numbers) >= 2:
                numbers.sort(reverse=True)
                pair_sum = numbers[0] + numbers[1]
                ans = max(ans, pair_sum)

        return ans

最后,看了灵佬的题解再次震撼我心,附上灵佬题解链接:

力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台

相关推荐
算AI4 小时前
人工智能+牙科:临床应用中的几个问题
人工智能·算法
我不会编程5555 小时前
Python Cookbook-5.1 对字典排序
开发语言·数据结构·python
owde6 小时前
顺序容器 -list双向链表
数据结构·c++·链表·list
第404块砖头6 小时前
分享宝藏之List转Markdown
数据结构·list
hyshhhh6 小时前
【算法岗面试题】深度学习中如何防止过拟合?
网络·人工智能·深度学习·神经网络·算法·计算机视觉
蒙奇D索大6 小时前
【数据结构】第六章启航:图论入门——从零掌握有向图、无向图与简单图
c语言·数据结构·考研·改行学it
A旧城以西6 小时前
数据结构(JAVA)单向,双向链表
java·开发语言·数据结构·学习·链表·intellij-idea·idea
杉之7 小时前
选择排序笔记
java·算法·排序算法
烂蜻蜓7 小时前
C 语言中的递归:概念、应用与实例解析
c语言·数据结构·算法
OYangxf7 小时前
图论----拓扑排序
算法·图论