刚面试完痛定思痛,开始肝哈希表第二天
梦回大一,想起自己算法从入门到放弃的拦路虎2333333
首先正常暴力需要两层循环,可以通过边遍历边查看历史记录来实现一层循环,恰好哈希表可以完美解决,因为是两个数,遍历过a时,存入target-a,再用于匹配后面的数
由于本题要返回对应的数组下标,因此使用map存储
class Solution {
public int[] twoSum(int[] nums, int target) {
int[] res = new int[2];
Map map = new HashMap<>();
for (int i = 0 ; i < nums.length; i++) {
if (map.containsKey(target - nums[i])){
res[0] = map.get(target - nums[i]);
res[1] = i;
break;
}
map.put(nums[i],i);
}
return res;
}
}
上面一个题的变种,四个数相加通过分治思想转化为两个两数之和再相加
感觉这题新手可能以为n平方时间复杂度肯定过不了,但题目有提示,所以直接大胆冲
1 <= n <= 200
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
Map map = new HashMap<>();
int count = 0;
for (int i = 0; i < nums1.length; i++){
for (int j = 0; j < nums2.length; j++) {
if (map.containsKey(nums1[i]+nums2[j])) {
map.put(nums1[i]+nums2[j],map.get(nums1[i]+nums2[j])+1);
}
else {
map.put(nums1[i]+nums2[j],1);
}
}
}
for (int i = 0; i < nums3.length; i++){
for (int j = 0; j < nums4.length; j++) {
if (map.containsKey(-nums3[i]-nums4[j])) {
count += map.get(-nums3[i]-nums4[j]);
}
}
}
return count;
}
}
看题解用的HashMap的方法,可以直接省去判断的过程,代码简化很多
map.getOrDefault
class Solution {
public int fourSumCount(int[] nums1, int[] nums2, int[] nums3, int[] nums4) {
int res = 0;
Map map = new HashMap();
//统计两个数组中的元素之和,同时统计出现的次数,放入map
for (int i : nums1) {
for (int j : nums2) {
int sum = i + j;
map.put(sum, map.getOrDefault(sum, 0) + 1);
}
}
//统计剩余的两个元素的和,在map中找是否存在相加为0的情况,同时记录次数
for (int i : nums3) {
for (int j : nums4) {
res += map.getOrDefault(0 - i - j, 0);
}
}
return res;
}
}
跟第一天的很像,也可以用数组来实现哈希表,这里练习下刚学的getOrDefault
class Solution {
public boolean canConstruct(String ransomNote, String magazine) {
Map map = new HashMap<>();
for (char c : magazine.toCharArray()) {
map.put(c,map.getOrDefault(c,0)+1);
}
for (char c : ransomNote.toCharArray()) {
map.put(c,map.getOrDefault(c,0)-1);
}
for (int i : map.values()) {
if (i < 0) return false;
}
return true;
}
}