LeetCode- 面試題56 - I. 數組中數字出現的次數

一個整型數組 nums 裏除兩個數字之外，其他數字都出現了兩次。請寫程序找出這兩個只出現一次的數字。要求時間複雜度是O(n)，空間複雜度是O(1)。

 

示例 1：

輸入：nums = [4,1,4,6]
輸出：[1,6] 或 [6,1]
示例 2：

輸入：nums = [1,2,10,4,1,4,3,3]
輸出：[2,10] 或 [10,2]
 

限制：

2 <= nums <= 10000
 

寫法有點耗時了。

#include <iostream>
#include <vector>
#include <unordered_map>
#include <set>
using namespace std;

class Solution {
public:
    vector<int> singleNumbers(vector<int>& nums) {
        vector<int>res;
        for(int i=0;i<nums.size();i++){
            m_map[nums[i]]++;
            m_set.insert(nums[i]);
        }
        set<int>::iterator it=m_set.begin();
        while(it!=m_set.end()){
            if(m_map[*it]==1){
                res.push_back(*it);
            }
            it++;
        }
        return res;
    }
private:
    unordered_map<int,int> m_map;
    set<int> m_set;
};

int main(){
    Solution *ps =new Solution();
    vector<int> nums={4,1,4,6};
    vector<int> nums2={1,2,10,4,1,4,3,3};
    vector<int> res = ps->singleNumbers(nums2);

    for(int i=0;i<res.size();i++){
        cout<<res[i]<<endl;
    }
    return 0;
}