2017-3-4 leetcode 414

虽说周末要早起来着，但是日子过得有点奇怪，一不小心就忘掉了。。。

leetcode414 https://leetcode.com/problems/third-maximum-number/?tab=Description

leetcode

leetcode

======================================

414说的是
给你n个数字（无序，有重复），输出第三大的数，如果没有，输出最大的，使用O(n)的算法。注意，你的排序中不能有相同的，也就是说如果输入2，3，2，3，我们要输出3，因为没有第三大的数字。

我的思路
这题目，一下子就会想到堆priority_queue，但是堆是nlogn的，好吧，手动维护三个数字分别表示最大、次大、次次大好了，但是初始化设成什么值好呢，用0x80 memset一下好了，得到的是-1e9，感觉 够大了。

 1 class Solution {
 2 public:
 3     int thirdMax(vector<int>& nums) {
 4         int aim[4];
 5         int n=nums.size();
 6         memset(aim,0x80,sizeof(aim));
 7         for(int i=0;i<n;i++){
 8             if(nums[i]>aim[0]){
 9                 aim[2]=aim[1];
10                 aim[1]=aim[0];
11                 aim[0]=nums[i];
12             }
13             if(aim[0]>nums[i]&&nums[i]>aim[1]){
14                 aim[2]=aim[1];
15                 aim[1]=nums[i];
16             }
17             if(aim[1]>nums[i]&&nums[i]>aim[2]){
18                 aim[2]=nums[i];
19             }
20         }
21         return aim[2]==aim[3]?aim[0]:aim[2];
22     }
23 };

WA

结果提交了一次，返回了WA，人家有数据是-2e9的。。。。好吧，那就写标记，用给的元素维护赋初值好了。

 1 class Solution {
 2 public:
 3     void swap(int &x,int &y){
 4         x^=y;y^=x;x^=y;
 5     }
 6     int thirdMax(vector<int>& nums) {
 7         int aim[3],flag=0;
 8         int n=nums.size();
 9         memset(aim,0x80,sizeof(aim));
10         for(int i=0;i<n;i++){
11             if(flag==0){
12                 aim[0]=nums[i];
13                 flag++;
14                 continue;
15             }
16             if(flag==1){
17                 if(nums[i]==aim[0])continue;
18                 aim[1]=nums[i];
19                 if(aim[0]<aim[1])swap(aim[0],aim[1]);
20                 flag++;
21                 continue;
22             }
23             if(flag==2){
24                 if(nums[i]==aim[0]||nums[i]==aim[1])continue;
25                 if(nums[i]>aim[0]){
26                     aim[2]=aim[1];
27                     aim[1]=aim[0];
28                     aim[0]=nums[i];
29                 }else if(aim[0]>nums[i]&&nums[i]>aim[1]){
30                     aim[2]=aim[1];
31                     aim[1]=nums[i];
32                 }else aim[2]=nums[i];
33                 flag++;
34                 continue;
35             }
36             if(nums[i]>aim[0]){
37                 aim[2]=aim[1];
38                 aim[1]=aim[0];
39                 aim[0]=nums[i];
40             }
41             if(aim[0]>nums[i]&&nums[i]>aim[1]){
42                 aim[2]=aim[1];
43                 aim[1]=nums[i];
44             }
45             if(aim[1]>nums[i]&&nums[i]>aim[2]){
46                 aim[2]=nums[i];
47             }
48         }
49         return flag!=3?aim[0]:aim[2];
50     }
51 };

AC

完成是完成了，这也太丑了吧。。。去学习学习别人的写法。结果发现了这个。。。。

1 int thirdMax(vector<int>& nums) {
2     set<int> top3;
3     for (int num : nums)
4         if (top3.insert(num).second && top3.size() > 3)
5             top3.erase(top3.begin());
6     return top3.size() == 3 ? *top3.begin() : *top3.rbegin();
7 }

（很优美。。。。不想折叠），虽然set确实提供了logn的复杂度用来插入和删除，但是我们只需要前3大的数据，所以set的size只需要有3就好，log3算是常数.。。。。。。。心悦诚服

=======================================