codetop刷题笔记1——两数之和/iota/lambda表达式

1 题目

两数之和

题目：

给定一个整数数组 nums 和一个整数目标值 target，请你在该数组中找出 和为目标值 target 的那两个 整数，并返回它们的数组下标。

你可以假设每种输入只会对应一个答案。但是，数组中同一个元素在答案里不能重复出现。
你可以按任意顺序返回答案。

 

示例 1：
输入：nums = [2,7,11,15], target = 9
输出：[0,1]
解释：因为 nums[0] + nums[1] == 9 ，返回 [0, 1] 。
示例 2：
输入：nums = [3,2,4], target = 6
输出：[1,2]
示例 3：
输入：nums = [3,3], target = 6
输出：[0,1]

解题：从小到大排序得到索引(因为需要得到原位置)，然后用左右指针来解题。

代码：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        int len = nums.size();
        vector<int> v(len);
        iota(v.begin(), v.end(), 0);
        sort(v.begin(),v.end(),[&](const int& a, const int& b) {return (nums[a] < nums[b]);});

        int left = 0,right=len-1;
        vector<int> result(2);
        while(true){
            if(nums[v[left]]+nums[v[right]]==target){
                result[0]=v[left];
                result[1]=v[right];
                break;
            }
            else if(nums[v[left]]+nums[v[right]]>target){
                right--;
            }
            else{
                left++;
            }
        }
        return result;
    }
};

2 iota

iota(v.begin(), v.end(), k)：给size为n的vector赋初值，值的大小k~k+n-1。

3 lambda表达式

lambda表达式语法：
[capture list](parameters) mutable throw() -> return-type {statement}
capture list：捕获列表，用于捕获上下文变量，[&]表示以引用的方式捕获所有变量，[&a]表示以引用的方式捕获变量a，[=]表示以值的方式捕获所有变量，[a]表示以值的方式捕获变量a，[&,a]表示a为值形式其余为引用，[=,&a]表示a为引用其余为值。

sort(v.begin(),v.end(),[&](const int& a, const int& b) {return (nums[a] < nums[b]);});

这里使用了sort函数的第三个参数。引入了一个lambda函数，其中将上下文变量以引用的方式捕获，即nums。