給定一個整數數組 nums 和一個目標值 target,請你在該數組中找出和為目標值的那 兩個 整數,并返回他們的數組下標。
你可以假設每種輸入只會對應一個答案。但是,你不能重復利用這個數組中同樣的元素。
來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/two-sum
著作權歸領扣網絡所有。商業轉載請聯系官方授權,非商業轉載請注明出處。
c++暴力破解
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> 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;
}
}
}
}
};
python暴力破解
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
ans=[]
for i in xrange(nums.__len__()):
for j in xrange(nums.__len__()):
if(j>i and nums[i]+nums[j]==target):
ans.append(i)
ans.append(j)
return ans
pytnon使用map
class Solution(object):
def twoSum(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: List[int]
"""
ans=[]
map = {}
for i in range(nums.__len__()):
map[(target-nums[i])]=i
print map
for i in range(nums.__len__()):
if(map.get(nums[i]) and i !=map.get(nums[i]) ):
ans=[]
ans.append(i)
ans.append(map.get(nums[i]))
return ans
c++使用map v1.0
class Solution {
public:
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
map<int,int> temp;
for (int i=0;i<nums.size();i++)
{
temp[target-nums[i]]=i;
}
for (int i=0;i<nums.size();i++)
{
cout<<nums[i]<<endl;
if(temp.find(nums[i])!=temp.end()&&i!=temp[nums[i]]){//這個地方順序很重要,原先一直報錯,原因在于temp[nums[i]]若是不存在會自動插入
result.push_back(i);
result.push_back(temp[nums[i]]);
return result;
}
}
}
};
c++使用map v2.0
這個版本在一個n循環都不到的時間中就可以實現
vector<int> twoSum(vector<int>& nums, int target) {
vector<int> result;
map<int,int> temp;
for (int i=0;i<nums.size();i++)
{
if (temp.count(target-nums[i])&&i!=temp[target-nums[i]]){
result.push_back(temp[target-nums[i]]);//temp[target-nums[i]]肯定在前面
result.push_back(i); //i肯定在后面
return result;
}
temp[nums[i]]=i;
}
}
python使用dict 版本
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
tmp={}
result=[]
for i,number in enumerate(nums):
if(number in tmp):
result.append(tmp[number])
result.append(i)
break
else:
tmp[target-number]=i # 鍵存儲的是差值,值存儲第一個數序號
return result
代碼更簡潔的python3版本
class Solution:
def twoSum(self, nums: List[int], target: int) -> List[int]:
tmp={}
for i,number in enumerate(nums):
if(number in tmp):
return [tmp[number],i]
else:
tmp[target-number]=i # 鍵存儲的是差值,值存儲第一個數序號
java版本
class Solution {
public int[] twoSum(int[] nums, int target) {
HashMap<Integer,Integer> tmp=new HashMap(); // java版本主要是對map的操作不熟悉,太麻煩了,一定要IDE
for (int i=0;i<nums.length;i++){
if(tmp.containsKey(nums[i])) return new int[] {tmp.get(nums[i]),i};
else tmp.put(target-nums[i],i);
}
return null;
}
}
注意事項
1.使用map會使得效率提高很多,主要是特殊的數據結構使得查找的效率很高
2.java中的hashMap底層用了hash表,TreeMap用了紅黑樹,python中的dict{}應該是采用哈希表,最低能在O(1)時間內完成搜索,c++中map使用的應該是紅黑樹
3.temp[nums[i]]這里要注意,若是不存在的話會自動插入0,就是一個簡單的cout<<temp[4],都會添加一個新元素導致很隱秘的錯誤。。。。
參考資料
1.python的dict實現[http://www.lxweimin.com/p/02af9673ab34]
2.Java HashMap工作原理及實現
[http://yikun.github.io/2015/04/01/Java-HashMap%E5%B7%A5%E4%BD%9C%E5%8E%9F%E7%90%86%E5%8F%8A%E5%AE%9E%E7%8E%B0/]