1、題目
Given an array of integers, return indices of the two numbers such that they add up to a specific target.
You may assume that each input would have exactly one solution, and you may not use the same element twice.
Example:
Given nums = [2, 7, 11, 15], target = 9,
Because nums[0] + nums[1] = 2 + 7 = 9,
return [0, 1].
2、代碼實作
方法1:
通過2層循環解決,相當于把每個情況都便利出來,看是否有符合要求的沒?代碼如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
if (nums.empty() || (nums.size() == 0))
return result;
for (int i = 0; i < nums.size() ; ++i) {
for (int j = i + 1; j < nums.size(); ++j) {
if (nums[i] + nums[j] == target) {
result.push_back(i);
result.push_back(j);
return result;
}
}
}
return result;
}
};
你看上面不是說了嘛?you may not use the same element twice,特麼我還i < nums.size(),如果i = nums.size() - 1, j = nums.size() -1這個時候恰好等于target的話,i == j,是自己沒有注意,以後不要忘記,雖然通過程式了,但是呢?時間複雜度O(n*n),如果給我們的數組都是正數的沒有順序的,是以我麼可以限制一些元素循環,比如if(ums[i] > target) continue,修改代碼如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
if (nums.empty() || (nums.size() == 0))
return result;
for (int i = 0; i < nums.size() - 1; ++i) {
for (int j = i + 1; j < nums.size(); ++j) {
if (nums[i] > target)
continue;
if (nums[i] + nums[j] == target) {
result.push_back(i);
result.push_back(j);
return result;
}
}
}
return result;
}
};
方法2:
既然是從一個數組裡面找到2個正确資料和為目标資料,我們用逆向思維,用目标數減去一個正确資料,那麼差就是另外一個正确資料,如果減去的不是正确的資料,那結果肯定也不是正确的資料,我們可以把所有的資料裝在hashmap裡面,key為資料,value為下标,周遊每個資料,如果目标資料減去每個周遊的資料的差是hashmap裡面的key,就可以了,我們可以得到value,也就是index.這樣時間複雜度降低了,變成了O(n)
代碼如下:
//import java.util.HashMap;
public class Solution {
public int[] twoSum(int[] nums, int target) {
if (null == nums)
return null;
int[] result = new int[2];
HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int i = 0; i < nums.length; ++i) {
Integer value = map.get(nums[i]);
if (null == value)
map.put(nums[i], i);
Integer index = map.get(target - nums[i]);
if (null != index && index < i)
{
result[0] = index;
result[1] = i;
}
}
return result;
}
}
這裡為什麼不把 下标設定成key,nums[i]設定成value呢?如果這樣設定的話,我們每次put(i, nums[i]),我們target - nums[i]得到另外一個正确的值,但是我們程式需要的下表,我需要通過正确的nums[i],的到i,也就 是說通過value的到key,得不到吧,是以請注意,反過來就可以了。
特麼我這麼寫還通過不了,下面就不寫判斷條件了,
import java.util.HashMap;
public class Solution {
public int[] twoSum(int[] nums, int target) {
int[] result = new int[2];
HashMap<Integer, Integer> map = new HashMap<Integer, Integer>();
for (int i = 0; i < nums.length; ++i) {
Integer values = Integer.valueOf(nums[i]);
if (!map.containkey(values))
map.put(i, nums[i]);
Integer index = map.get(target - nums[i]);
if (null != index && index < i)
{
result[0] = index;
result[1] = i;
}
}
return result;
}
}
提示出現:
Line 9: error: cannot find symbol: method containkey(Integer)
明天eclipse跑下,日了狗
C++實作代碼如下:
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
if (nums.size() == 0 || nums.empty())
return result;
map<int, int> map;
for (int i = 0; i < nums.size(); ++i) {
if (!map.count(nums[i])) { //特麼我這裡忘記寫!,日了狗
map.insert(pair<int, int>(nums[i], i));
}
if (map.count(target - nums[i])) { //這裡自己傻逼了,寫成map[target - nums[i]]了
int index = map[target - nums[i]];
if (index < i) {
result.push_back(index);
result.push_back(i);
return result;
}
}
}
return result;
}
};
3、總結
1、寫代碼的時候要注意,如果有2個下标的話,第二個大需要j = i + 1,這樣就永遠大了,然後就是i 和 j不相等的話,注意i < nuns[i].size() - 1,不是i < nuns[i].size(),不這樣的話i和j有可能就相等了。
2、我們為了降低時間複雜度,我們可以逆向思路,既然2個數加起來為目标數,那麼目标數減去一個正确的資料,另外肯定是一個正确的資料,就可以通過map資料結構得到下标,
3、我們寫代碼的時候,一定要注意if(!map.count(nums[i])),不要寫成了if(map.count(nuts[i])),不要忘記了!.
4、熟悉了C++的map用法和count函數是否包含key的作用。