LeetCode：342. 数位和相等数对的最大和（C++）

目录

342. 数位和相等数对的最大和

题目描述：

实现代码与解析：

暴力模拟（超时，考虑优化）

如何优化？

hash

---

342. 数位和相等数对的最大和

题目描述：

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

实现代码与解析：

暴力模拟（超时，考虑优化）

class Solution {
public:
    int maximumSum(vector& nums) {
        int res = -1;
        for (int i = 0; i < nums.size() - 1; i++) {
            for (int j = i + 1; j < nums.size(); j++) {
                if (nums[i] + nums[j] < res) continue;
                int a = nums[i], b = nums[j];
                int suma = 0;
                int sumb = 0;
                while (a) {
                    suma += a % 10;
                    a /= 10;
                }
                while (b) {
                    sumb += b % 10;
                    b /= 10;
                }
                if (suma == sumb) {
                    res = max(res, nums[i] + nums[j]);
                }
            }
        }
        return res;
    }
};

如何优化？

        不难发现，在计算一个数的数位和，有的可能已经计算过了，不必再重复计算，我们考虑用哈希，或者数组存一下已经计算过的数的数位和即可，应该就能过了。

hash

class Solution {
public:
    int maximumSum(vector& nums) {
        int res = -1;

        unordered_map m; // 数位和，数字
        
        for (auto t: nums) {
            int tt = t;
            int sumt = 0;
            while(tt) { // 计算数位和
                sumt += tt % 10;
                tt /= 10;
            } 
            if (!m[sumt]) {
                m[sumt] = t; // 若没出现过，hash记录一下
            }
            else { // 若有已有相同数位和，计算，后与res比较大小
                res = max(res, m[sumt] + t);
                m[sumt] = m[sumt] > t ? m[sumt] : t; // 更新，保留大的
            }
        }
        return res;
    }
};

看注释即可。