lintcode 两数和-小于或等于目标值

描述

给定一个整数数组，找出这个数组中有多少对的和是小于或等于目标值。返回对数。

样例

例1:

输入: nums = [2, 7, 11, 15], target = 24.
输出: 5.
解释:
2 + 7 < 24
2 + 11 < 24
2 + 15 < 24
7 + 11 < 24
7 + 15 < 25
例2:

输入: nums = [1], target = 1.
输出: 0.

思考

先排序
使用两个指针，从首尾开始，如果和小于等于目标值的话，那么从j到i+1之间的 和i的和 都符合条件，
如果大于目标值，那么i++

代码

class Solution {
public:
    /**
     * @param nums: an array of integer
     * @param target: an integer
     * @return: an integer
     */
    int twoSum5(vector<int> &nums, int target) {
        // write your code here
        if (nums.size() == 0) return 0;
        if (nums.size() == 1) return target >=2 * nums[0]?1:0;
        sort(nums.begin(), nums.end());
        
        int res = 0;
        int i = 0, j = nums.size() - 1;
        while (i < j) {
            if (nums[i] + nums[j] <= target) {
                res += j-i;
                i++;   
            }
            else
                j--;
        }
        
        return res;
    }
};