229. 多数元素 II

Posted 2022-10-03 心脏dance

题目链接：力扣

思路：数据大于floor(n/3)向下取整的数最多只有两个（仔细想想），或者1个，或者没有。这样的话就采取类似最近很火的游戏“羊了个羊”消除，当然区别就是我们槽位只有2个，点击第三个的时候只有不一样这三个才会同时消除，如果相同就叠在对应的槽位上，这样最后剩下的两个槽位上面的值才有可能大于floor(n/3)，为什么说有可能，因为存在一种特殊情况：【3，2，3】，这种情况下通过我们的算法最后会出现，第一个槽位剩下3，第二个槽位剩下2，但是只有3是满足的，所以最后我们再把这两个值验证一下即可。别的能消除的一定不会大于floor(n/3)，因为消除的是三个不一样的数。

两次for循环，所以时间复杂度为O(n)。

空间上面我们只开了几个变量，所以时间复杂度O(1)。

上代码：

class Solution 
    fun majorityElement(nums: IntArray): List<Int> 
        val count = nums.size / 3
        var numOne: Int? = null
        var numTwo: Int? = null
        var numOneCount = 0
        var numTwoCount = 0
        nums.forEach  x ->
            when (x) 
                numOne -> numOneCount++
                numTwo -> numTwoCount++
                else -> 
                    when (null) 
                        numOne -> 
                            numOne = x
                            numOneCount++
                        
                        numTwo -> 
                            numTwo = x
                            numTwoCount++
                        
                        else -> 
                            numOneCount--
                            if (numOneCount == 0) numOne = null
                            numTwoCount--
                            if (numTwoCount == 0) numTwo = null
                        
                    
                
            
        
        numOneCount = 0
        numTwoCount = 0
        nums.forEach  x ->
            when 
                numOne != null && x == numOne -> numOneCount++
                numTwo != null && x == numTwo -> numTwoCount++
            
        
        val list = mutableListOf<Int>()
        if (numOneCount > count) numOne?.let  list.add(it) 
        if (numTwoCount > count) numTwo?.let  list.add(it) 
        return list