Day Four 第三大數Leetcode 414題

題目如下：

給定一個非空數組，返回此數組中第三大的數。如果不存在，則返回數組中最大的數。要求算法時間複雜度必須是O(n)。

示例 1:

輸入: [3, 2, 1]

輸出: 1

解釋: 第三大的數是 1.

示例 2:

輸入: [1, 2]

輸出: 2

解釋: 第三大的數不存在, 所以返回最大的數 2 .

示例 3:

輸入: [2, 2, 3, 1]

輸出: 1

解釋: 注意，要求返回第三大的數，是指第三大且唯一出現的數。
存在兩個值爲2的數，它們都排第二。

代碼如下：

class Solution{
	public:
		int thirdMax(vector<int>&nums){
			int NumOne=nums[0],NumTwo=nums[0],NumThree=nums[0];
			int cont=0,set=0;
			for(int i=1;i<nums.size();i++){
				if(nums[i]>NumOne){
					NumOne=nums[i];
				}else if(nums[i]!=NumOne){
					NumTwo=nums[i];
				}else{
					cont++;
				}
			}
			if(cont==nums.size-1){
				return NumOne;
			}else{
				for(int i=1;i<nums.size;i++){
				   if(nums[i]>NumTwo&&nums[i]!=NumOne){
					   
					   NumTwo=nums[i];
				}else if(nums[i]<NumTwo){
					   NumThree=nums[i];
				}else if(nums[i]==NumTwo){
					set++;
				}
		    }
		    if(cont+set==nums.size-1){
		    	return NumTwo;
			}else{
				for(int i=1;i<nums.size;i++){
				   if(nums[i]>NumTwo&&nums[i]!=NumOne&&nums[i]!=NumTwo){
					   NumThree=nums[i];
				}
				}
				return NumThree;	
			}
		    		
			}		
			
		}
}; 

             
            }
          }
       
        }
};