leetcode Two Sum hash

传送门:Two Sum

给定一个数组,求出其中和为target的两个数的位置,输入保证只有一组解

用map记录每个数出现的位置,然后对于数组中的数x,查找target - x是否出现过,出现则返回对应两个数的位置,否则记录x的出现的位置

class Solution {
public:
    vector<int> twoSum(vector<int> &numbers, int target) {
        vector <int> ans;
        map <int,int> p;
        for(int i = 0;i < numbers.size();i++)
        {
            int x = numbers[i];
            int y = target - x;
            if(p[y] != 0)
            {
                ans.push_back(p[y]);
                ans.push_back(i+1);
                return ans;
            }
            else
                p[numbers[i]] = i + 1;
        }
    }
};

用dictionary记录每个数的出现的位置，然后同上

class Solution:
    # @return a tuple, (index1, index2)
    def twoSum(self, num, target):
        p = {}
        cnt = 0
        for x in num:
            y = target - x
            if p.has_key(y):
                return p[y],cnt+1
            else:
                p[x] = cnt + 1
            cnt = cnt + 1