題目:給定一個(gè)整數(shù)數(shù)組 nums 和一個(gè)目標(biāo)值 target,請(qǐng)你在該數(shù)組中找出和為目標(biāo)值的那 兩個(gè) 整數(shù),并返回他們的數(shù)組下標(biāo)。
你可以假設(shè)每種輸入只會(huì)對(duì)應(yīng)一個(gè)答案。但是,你不能重復(fù)利用這個(gè)數(shù)組中同樣的元素。
來(lái)源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/two-sum
著作權(quán)歸領(lǐng)扣網(wǎng)絡(luò)所有。商業(yè)轉(zhuǎn)載請(qǐng)聯(lián)系官方授權(quán),非商業(yè)轉(zhuǎn)載請(qǐng)注明出處。
方法一:很容易想到的暴力法,雙重for循環(huán),遍歷2遍數(shù)組,窮舉法找到和為target的兩數(shù)數(shù)組下標(biāo)。
代碼如下:
public static int[] fun1(int[] nums, int target) {
for (int i = 0; i < nums.length; i++) {
for (int j = i + 1; j < nums.length; j++) {
// 滿足條件則返回
if (nums[i] + nums[j] == target) {
return new int[] {i,j};
}
}
}
// 代碼的魯棒性體現(xiàn),未找到結(jié)果則拋出異常
throw new IllegalArgumentException("no two nums match");
}
方法二:一遍哈希法。首先創(chuàng)建一個(gè)哈希表,然后遍歷數(shù)組,如果將當(dāng)前正在被遍歷的元素稱為num1,另外一個(gè)滿足條件的數(shù)稱為num2,則有num2 = target - num1。遍歷當(dāng)前元素時(shí),只需查看哈希表中是否存在num2,如果存在,則找到了和為target的兩個(gè)數(shù)。如果不存在,將當(dāng)前正在遍歷的元素的值作為key,下標(biāo)索引作為value存入哈希表,繼續(xù)遍歷下一個(gè)。
代碼如下:
public static int[] fun1(int[] nums, int target) {
Map<Integer,Integer> map = new HashMap<>();
for (int i = 0;i < nums.length;i++) {
// 待尋找的第二個(gè)數(shù)
int num2 = target - nums[i];
// 如果哈希表中存在第二個(gè)數(shù)則找到了滿足條件的
if (map.containsKey(num2)) {
return new int[] {map.get(num2),i};
}
// 否則未找到將當(dāng)前元素放入哈希表,將值作為key,下標(biāo)索引作為value
map.put(nums[i],i);
}
// 魯棒性
throw new IllegalArgumentException("no two nums match");
}
自測(cè)驗(yàn)證,也可直接在LeetCode上提交驗(yàn)證。
public static void main(String[] args) {
int[] nums = {2, 7, 11, 15};
int target = 9;
int[] result = fun2(nums,target);
// 輸出的數(shù)組下標(biāo)應(yīng)該是0和1
Arrays.stream(result).forEach(System.out::println);
}