面试题56 - I. 数组中数字出现的次数(java)

面试题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

---

解法一：分组异或

• 思路
  • 将两个不同的数字分为不同组
  • 将相同的数字分在同一组
• 步骤
  • 对全体异或，得到一个异或合成词
  • 用&提取其中最后一位1所在位置作为标记（由于1是由0^1实现，即可将两个不同数字分成两组）
  • 通过标记值&分组异或

class Solution {
    public int[] singleNumbers(int[] nums) {
        int k=0;
        //得到合成词
        for(int num:nums){
            k^=num;
        }
        int mask=1;
        //找出分组标记
        while((k&mask)==0){
            mask<<=1;
        }
        int a=0,b=0;
        for(int num:nums){
            //用分组标记分组异或
            if((num&mask)==0){
                a^=num;
            }
            else{
                b^=num;
            }
        }
        return new int[]{a,b};
    }
}