1. 程式人生 > 實用技巧 >java 集合List、Set

java 集合List、Set

題目描述

輸入一個遞增排序的陣列和一個數字s,在陣列中查詢兩個數,使得它們的和正好是s。如果有多對數字的和等於s,則輸出任意一對即可。

示例1:

輸入:nums = [2,7,11,15], target = 9
輸出:[2,7] 或者 [7,2]

示例2:

輸入:nums = [10,26,30,31,47,60], target = 40
輸出:[10,30] 或者 [30,10]

限制:

1 <= nums.length <= 10^5
1 <= nums[i] <= 10^6

來源:力扣(LeetCode)
連結:https://leetcode-cn.com/problems/he-wei-sde-liang-ge-shu-zi-lcof

程式碼實現

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        int i, j;
        vector<int> result(2, 0);
        i = 0; j = nums.size() - 1;
        while(i < j) {
            int sum = nums[i] + nums[j];
            if(sum > target)
                j--;
            else if(sum < target)
                i++;
            else{
                result[0] = nums[i];
                result[1] = nums[j];
                return result;
            }
        }
        return result;
    }
};