目录
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<int>& 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<int>& nums) {int res = -1;unordered_map<int, int> 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;}
};
看注释即可。