LeetCode 2342.数位和相等数对的最大和
给你一个下标从 0 开始的数组 nums ,数组中的元素都是 正 整数。请你选出两个下标 i 和 j(i != j),且 nums[i] 的数位和 与 nums[j] 的数位和相等。
请你找出所有满足条件的下标 i 和 j ,找出并返回 nums[i] + nums[j] 可以得到的 最大值。如果不存在这样的下标对,返回 -1。
示例 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^55
1 <= nums[i] <= 109^99
枚举nums,记录之前数字的数位和及该数位和对应的最大值:
class Solution {
public:int maximumSum(vector<int>& nums) {unordered_map<int, int> sumMax;int ans = -1;for (int i : nums) {int digitSum = getDigitSum(i);if (sumMax.find(digitSum) == sumMax.end()) {sumMax[digitSum] = i;} else {ans = max(ans, sumMax[digitSum] + i);sumMax[digitSum] = max(sumMax[digitSum], i);}}return ans;}int getDigitSum(int num) {int ans = 0;while (num) {ans += num % 10;num /= 10;}return ans;}
};
如果nums的长度为n,其中数字的大小为m,则此算法时间复杂度为O(nlogm);空间复杂度为O(logm),因为数字的位数是logm级别的,而数位和的种类数是9*logm,即如果数字有3位,那么其数位和的取值范围就是0(全0)~27(全9)。
