1. 程式人生 > >LeetCode 刷題1---兩數之和

LeetCode 刷題1---兩數之和

/**
給定一個整數陣列 nums 和一個目標值 target,請你在該陣列中找出和為目標值的那 兩個 整數,並返回他們的陣列下標。
你可以假設每種輸入只會對應一個答案。但是,你不能重複利用這個陣列中同樣的元素。
示例:
給定 nums = [2, 7, 11, 15], target = 9
因為 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]
*/

public class demo1 {
//普通解法
public static int[] twoSum(int[] nums,int target){
HashMap<Integer,Integer> m = new HashMap<Integer, Integer>();
int res[] = new int[2];
for(int i=0;i<nums.length;++i){
if(m.containsKey(target-nums[i])){
res[0]=i;
res[1]=m.get(target-nums[i]);
break;
}
m.put(nums[i],i);
}
return res;
}
//二分法解決方案
/**
* 建立一個輔助類陣列,對輔助類進行排序,使用兩個指標,開始時分別指向陣列的兩端,
* 看這兩個下標對應的值是否等於目標值,如果等於就從輔助類中找出記錄的下標,構造好返回結果,返回。
* 如果大於就讓右邊的下標向左移,進入下一次匹配,如果小於就讓左邊的下標向右移動,進入下一次匹配,
* 直到所有的資料都處理完
*/
    /**
* 輔助類
*/
private static class Node implements Comparable<Node> {
int val; // 值
int idx; // 值對應的陣列下標

public Node() {
}

public Node(int val, int idx) {
this.val = val;
this.idx = idx;
}

// 比較方法
public int compareTo(Node o) {
if (o == null) {
return -1;
}
return this.val - o.val;
}
}
public static int[] twoSum2(int[] nums, int target) {
// 用於儲存返回結果
int[] result = {0, 0};

// 建立輔助陣列
Node[] tmp = new Node[nums.length];
for (int i = 0; i < nums.length; i++) {
tmp[i] = new Node(nums[i], i);
}

// 對輔助陣列進行排序
Arrays.sort(tmp);

// 記錄輔助陣列中左邊一個值的下標
int lo = 0;
// 記錄輔助陣列中右邊一個值的下標
int hi = nums.length - 1;

// 從兩邊向中間靠隴進行求解
while (lo < hi) {
// 如果找到結果就設定返回結果,並且退出迴圈
if (tmp[lo].val + tmp[hi].val == target) {

if (tmp[lo].idx > tmp[hi].idx) {
result[0] = tmp[hi].idx + 1;
result[1] = tmp[lo].idx + 1;
} else {
result[0] = tmp[lo].idx + 1;
result[1] = tmp[hi].idx + 1;
}
break;
}
// 如果大於目標值,右邊的下標向左移動
else if (tmp[lo].val + tmp[hi].val > target) {
hi--;
}
// 如果小於目標值,左邊的下標向右移動
else {
lo++;
}
}
return result;
}
public static void main(String args[]) {
long startTime = System.currentTimeMillis();
int arr[]={5,6,7,8,9,10,1,12,3,4,2,3,4,5,7,11};
int target=20;
int res[]=twoSum(arr,target);
for(int i=0;i<res.length;++i){
System.out.println(res[i]);
}
long endTime = System.currentTimeMillis();
System.out.println("程式執行時間:" + (endTime - startTime) + "ms");
}
}