菜鸟LEETCODE刷题之路:1.两数之和

最近时间比较宽裕,准备每天刷几道题,练练脑子,刷的每道题咋这里记录一下。

1.两数之和

给定一个整数数组和一个目标值,找出数组中和为目标值的两个数。

你可以假设每个输入只对应一种答案,且同样的元素不能被重复利用。

我的代码:

class Solution {
public:
vector twoSum(vector& nums, int target) {
    vector r;
    vector::iterator iter1,iter2;
    int count1=-1;
    for(iter1=nums.begin();iter1!=nums.end();iter1++)
    {
        count1+=1;
        int count2=count1;
        for(iter2=iter1+1;iter2!=nums.end();iter2++)
        {
            count2+=1;
            if(*iter1+*iter2==target)
            {
                r.push_back(count1);
                r.push_back(count2);
                return r;
            }
        }
    }
    return r;
}

};

时间复杂度n*n,用时168ms,效率较低,效率较高的解法如下:

class Solution {
public:
    vector twoSum(vector& nums, int target) {
        vector result;
        int sz = nums.size();
        int left = 0;
        int right = sz - 1;
        
        vector sorted = nums;
        std::sort(sorted.begin(), sorted.end());
        while (left < right) {
            int sum = sorted[left] + sorted[right];
            if (sum == target) {
                for (int i = 0; i < sz; i++) {
                    if (nums[i] == sorted[left]) {
                        result.push_back(i);
                    }
                    else if (nums[i] == sorted[right]) {
                        result.push_back(i);
                    }
                    if (result.size() == 2) {
                        return result;
                    }
                    
                }
            }
            else if (sum > target) {
                right--;
            }
            else {
                left++;
            }
        }
        
        
        return result;
        
    }

};

此种解法将原数列进行排序后,再通过比较排序后数列的首尾之和与目标的大小从而调整首尾的位置来解决,用时4ms,时间复杂度为nlogn。

笔记:std内置排序算法std::sort

你可能感兴趣的:(菜鸟LEETCODE刷题之路:1.两数之和)