LeetCode 137. Single Number II

 

题目来源：https://leetcode.com/problems/single-number-ii/

问题描述

137. Single Number II

Medium

Given a non-empty array of integers, every element appears three times except for one, which appears exactly once. Find that single one.

Note:

Your algorithm should have a linear runtime complexity. Could you implement it without using extra memory?

Example 1:

Input: [2,2,3,2]

Output: 3

Example 2:

Input: [0,1,0,1,0,1,99]

Output: 99

------------------------------------------------------------

题意

给定一个数组，除了一个元素只出现了一次以外，其他元素都出现了3次，求只出现了一次的那个元素。要求算法的时间复杂度O(n)，空间复杂度O(1).

------------------------------------------------------------

思路

使用自动机的思想。需要构造一个有限状态自动机，不断接受数组的输入，并且某一个状态可以消去3个相同的元素，保留单个元素。由于可以接受3个输入，因此自动机的状态至少有3个，需要2 bit的数据来表示状态，不妨设两个bit的状态为(a, b)。

考察输入数组元素的每个bit，不妨记作x，则根据“消去3个1，保留单个1”的原则，我们可以得到下面的状态转移表（其中(a’, b’)表示状态(a, b)接收到输入x后转移的状态）：

a	b	x	a’	b’
0	0	1	0	1
0	1	1	1	0
1	0	1	0	0
0	0	0	0	0
0	1	0	0	1
1	0	0	1	0

任何一个满足上表的布尔式都符合要求，最终结果即那个只出现一次的元素存储在状态量b中。

------------------------------------------------------------

代码

class Solution {
    public int singleNumber(int[] nums) {
        int a = 0, b = 0;
        for (int num: nums) {
            b = (b ^ num) & (~a);
            a = (a ^ num) & (~b);
        }
        return b;
    }
}