LeetCode_2342_数位和相等数对的最大和

题目链接

• https://leetcode.cn/problems/max-sum-of-a-pair-with-equal-sum-of-digits/

题目描述

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

请你找出所有满足条件的下标 i 和 j ，找出并返回 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 <= 105
• 1 <= nums[i] <= 109

解题思路

• 首先对数组进行排序，然后写一个函数获得每个整数的各个数位和
• 初始最大值是-1，遍历数组，使用哈希表记录，每次如果更新数位和时，就可以直接放入当前的nums[i]，来确保放入的和得到的都是当前最大的

AC代码

class Solution {
    public int maximumSum(int[] nums) {
        Arrays.sort(nums);
        int ans = -1;
        HashMap<Integer, Integer> map = new HashMap<>();
        for (int i = 0; i < nums.length; i++) {
            int k = getSum(nums[i]);
            if (map.containsKey(k)) {
                ans = Math.max(nums[i] + map.get(k), ans);
            }
            map.put(k, nums[i]);
        }
        return ans;
    }

    private static int getSum(int n) {
        int sum = 0;
        while (n > 0) {
            int tmp = n % 10;
            sum += tmp;
            n = n / 10;
        }
        return sum;
    }
}