leetcode1. Two Sum

給定一個整數數組 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/]

最后編輯于
?著作權歸作者所有,轉載或內容合作請聯系作者
平臺聲明:文章內容(如有圖片或視頻亦包括在內)由作者上傳并發布,文章內容僅代表作者本人觀點,簡書系信息發布平臺,僅提供信息存儲服務。

推薦閱讀更多精彩內容