Skip to content

Latest commit

 

History

History

480.Sliding-Window-Median

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 

480.Sliding-Window-Median

希望能够自动排序的容器,首选set或者multiset。对于这类非线性的容器,不能用+k或-k来实现迭代器的移动,必须用for循环不断++/--,或者使用next或prev。例如

set<int>::iterator pos = next(Set.begin(),k); //从Set.begin()开始后移k个
set<int>::iterator pos = prev(Set.end(),k);  //从Set.end()开始前移k个

本题里设置一个multiset,初始化时用next命令把pos定位到最中间(Set容量为奇数)、或者中间靠右(Set容量为偶数)的那个迭代器。

每次一进一出的操作如下。对于k奇偶性通用的代码比较难理解,那就索性分别讨论容易理清思路。

        for (int i=k; i<=nums.size(); i++)
        {
            if (k%2==1) results.push_back(*pos);
            else results.push_back(*pos/2.0+*prev(pos,1)/2.0);
            
            if (i==nums.size()) break;
            
            Set.insert(nums[i]);
            
            if (k%2==1)   // 当k为奇数时,Set内的排列为 OXO,X为pos指针
            {
                if (nums[i]>=*pos)
                    pos++;
                                
                // 加入一个元素后,此时阵型为 OOXO
                if (nums[i-k]>*pos)
                    pos--;
                else if (nums[i-k]<*pos)
                    ;
                else if (nums[i-k]==*pos && *prev(pos,1)!=*pos)  
                    // 如果X左边是个重复元素,则无需改动;如果X左边不是重复元素,则X会被删除,指针需左移
                    pos--;                
            }
            else    // 当k为偶数时,Set内的排列为 OOXO,X为pos指针
            {
                if (nums[i]<*pos)
                    pos--;
                
                // 加入一个元素后,此时阵型为 OOXOO
                if (nums[i-k]<*pos)
                    pos++;
                else if (nums[i-k]>*pos)
                    ;
                else if (nums[i-k]==*pos)
                    // 如果X左边是个重复元素,则会删除那个O;如果X左边不是重复元素,则会删除X。两种情况下指针都需右移
                    pos++;
            }
            
            Set.erase(Set.lower_bound(nums[i-k]));     
        }

每次内循环之后,pos的定义不变(最中间或者最中间靠右的迭代器)。

Leetcode Link