力扣260-只出现一次的数字-C++

一、运行结果

 

二、题目

给定一个整数数组 nums，其中恰好有两个元素只出现一次，其余所有元素均出现两次。 找出只出现一次的那两个元素。你可以按 任意顺序 返回答案。

进阶：你的算法应该具有线性时间复杂度。你能否仅使用常数空间复杂度来实现？

示例 1：

输入：nums = [1,2,1,3,2,5]
输出：[3,5]
解释：[5, 3] 也是有效的答案。
示例 2：

输入：nums = [-1,0]
输出：[-1,0]
示例 3：

输入：nums = [0,1]
输出：[1,0]

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/single-number-iii

三、思路

首先对数组进行排序，则相同的数字必定排列在一起，在找到第一个仅出现一次的数字前，只考虑偶数下标的数字（0, 2, 4, 6 ····）如果当前数字和它的下一个数字不等，那么该数字就是仅出现一次的数字。找到第一个之后，从下一个位置开始继续往后找，只考虑奇数下标的数字，如果当前数字和其下一个数字不等，那么该数字就是仅出现一次的数字。如果遍历到数组倒数第二个数字时还是只能找到一个仅出现一次的数字（最后一个数字不能和下一个数字比较），那么第二个仅出现一次的数字就是最后位置的数字。

四、代码

class Solution {
public:
    vector<int> singleNumber(vector<int>& nums) {
        int len = nums.size();
        sort(nums.begin(), nums.end());
        vector<int> ans;
        int index = 0;
        while(index < len-1){
            if(nums[index] != nums[index+1]){
                ans.push_back(nums[index]);
                index++;  //从第一个的下一个继续往后寻找第二个
                if(ans.size() == 2) return ans;
            }
            else index += 2;
        }
        if(ans.size() < 2) ans.push_back(nums[len-1]); //有可能最后一个元素只出现了一次
        return ans;
    }
};