给你一个下标从 0 开始的数组 nums ,数组中的元素都是正整数。请你选出两个下标 i 和 j(i != j),且 nums[i] 的数位和与nums[j]的数位和相等。请你找出所有满足条件的下标 i 和 j ,找出并返回 nums[i] + nums[j] 可以得到的最大值。如果不存在两个数位和相同的数,返回 −1。
思路:对于这种一左一右两个数的题目,通用套路是【维护左,枚举右】,用哈希表或者数组mx[s]来记录数位和s对应的最大的nums[i],用mx[s]+nums[j] 更新答案的最大值。
public int maximumSum(int[] nums) {
int ans = -1;
int[] mx = new int[82]; // 至多 9 个 9 相加
for (int num : nums) {
int s = 0; // num 的数位和
for (int x = num; x > 0; x /= 10) { // 枚举 num 的每个数位
s += x % 10;
}
if (mx[s] > 0) { // 说明左边也有数位和等于 s 的元素
ans = Math.max(ans, mx[s] + num); // 更新答案的最大值
}
mx[s] = Math.max(mx[s], num); // 维护数位和等于 s 的最大元素
}
return ans;
}
下面用HashMap实现一下:
思路:先对数组进行升序排序,从后往前循环,将nums[i]的数位和作为map的key,nums[i]作为key所对应的值。
public static int maximumSum1(int[] nums) {
Arrays.sort(nums);
int ans = -1, n = nums.length;
Map<Integer, Integer> map = new HashMap<>();
for (int i = n - 1; i >= 0; i--) {
int sum = 0;
for (int x = nums[i]; x > 0; x /= 10)
sum += x % 10;
if (map.containsKey(sum))
ans = Math.max(ans, map.get(sum) + nums[i]);
else map.put(sum, nums[i]);
}
return ans;
}