LeetCode 2342. 数位和相等数对的最大和 哈希

给你一个下标从 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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/max-sum-of-a-pair-with-equal-sum-of-digits
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。.

最初的想法：

枚举，两个和两个找是n^2，然后求各位的数字之和时间为logn，总的时间复杂度是O(n ^ 2logn)。所以要优化。

优化

对于数字之和，使用哈希表维护一个可以家出来他的最大值。然后从小到大枚举nums，更新结果
时间复杂度优化到O(nlogn)

class Solution {
public:
    int maximumSum(vector<int>& nums) {
        int res = -1;
        unordered_map<int, int> hash;
        for(auto x: nums){
            int s = 0, y = x;
            while(y) s += y % 10, y /= 10;
            // 如果hash里边没有这个索引，默认返回0
            if(hash.count(s)) res = max(res, x + hash[s]);
            hash[s] = max(hash[s], x);
        }

        return res;
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16