[LeetCode]Two Sum

2016-10-31  本文已影响40人  肖浩呗

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.

Example:

Given nums = [2, 7, 11, 15], target = 9

Because nums[0] + nums[1] = 2 + 7 = 9,return [0, 1].


Answer:

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        vector<int> result;
        int m_count = nums.size();
        int m_target = target;
        
        if (m_count < 2){
            return result;
        }
        
        if (m_count == 2){
            if((nums[0] + nums[1]) == target)
            {
                result.push_back(0);
                result.push_back(1);
                return result;
            }
            else{
                return result;
            }
        }
        
        int m_min = nums.at(0);
        int m_max = nums.at(0);
        
        for (int index = 1;index < m_count;index++){
            if(nums[index] > m_max){
                m_max = nums[index];
                continue;
            }
            if(nums[index] < m_min){
                m_min = nums[index];
            }
        }
        
        map<int,map<string,int>> m_flagMap;
        
        for (int index = 0;index < m_count;index++){
            m_flagMap[nums[index]]["bool"] = 1;
            m_flagMap[nums[index]]["count"] ++;
            m_flagMap[nums[index]]["index"] = index;
        }
        
        for (int index = 0;index < m_count;index++){
            int m_sub = m_target - nums[index];
            map<string,int> indexMap = m_flagMap[m_sub];
            if(indexMap["bool"] == 1 && indexMap["index"] != index){
                result.push_back(index);
                result.push_back(indexMap["index"]);
                break;
            }
        }
        
        return result;
    }
};
上一篇下一篇

猜你喜欢

热点阅读