线性时间常数空间找到数组中数目超过n/5的所有元素

问题描述：

Design an algorithm that, given a list of n elements in an array, finds all the elements that appear more than n/3 times in the list. The algorithm should run in linear time ( n >=0 )
You are expected to use comparisons and achieve linear time. No hashing/excessive space/ and don't use standard linear time deterministic selection algo

 

解答：

借鉴俄罗斯方块的玩法，先给个直观的例子。对于数组 4 3 3 2 1 2 3 4 4 7，按照类俄罗斯方块的玩法，当7要往下落的时候，屏幕上会呈现这个场景

4
4 3 2
4 3 2 1 _

当7落下时，获得了

4
4 3 2
4 3 2 1 7

此时最后一行元素数目满了（最后一行的大小为5），将其删除，继续落数字。直到所有数字都落下来了。此时数目超过n/5的元素一定在最后一行（因为假设数字a的个数超过n/5，删除操作最多执行了n/5次，所以最后必然a还在最后一行中）。但是，最后一行中的元素也有可能不是我们要的。此时需要扫描一下来确定。

同样的算法对于任意的m（m为正整数）均适用。

 

给出代码：

#include <iostream>
#include <map>
#include <algorithm>
typedef std::map<int, int> Map;
 Map findOverNth(int arr[], int size, int n)
{
    Map ret_map;
    typedef Map::value_type Elem; //pair<CONST int, int>
    int total = 0;
    std::for_each(arr, arr + size, [&, n](int val)
    {
        auto ret_pair = ret_map.insert(Elem(val, 0));
        ++(*ret_pair.first).second; ++ total;
        if (ret_map.size() == n)
            for (auto iter = ret_map.begin(); iter != ret_map.end(); )
            {
                --(*iter).second; -- total;
                if ((*iter).second == 0)
                    ret_map.erase(iter++);
                else
                    iter++;
            }
    });
    std::for_each(ret_map.begin(), ret_map.end(), [](Elem &elem) {elem.second = 0;});
    std::for_each(arr, arr + size, [&ret_map](int val) {if (ret_map.find(val) != ret_map.end()) ret_map[val] ++;});
    for (auto iter = ret_map.begin(); iter != ret_map.end(); )
    {
        if ((*iter).second <= size / n)
            ret_map.erase(iter++);
        else
            iter++;
    }
    return ret_map;
}
using namespace std;
int main()
{
    //int arr[] = {5,6,7,8, 10, 4,4, 4, 4,1, 1,1};
    int arr[] = {5,6,7,8, 10, 10, 10,10,10,10, 4,4, 4, 4,4,1, 1,1,1};
    auto a_map = findOverNth(arr, sizeof(arr)/sizeof(int), 4);
    cout<<sizeof(arr)/sizeof(int)<<endl;
    //cout<<a_map.size()<<endl;
    for each(auto elem in a_map)
    {
        cout<<elem.first<<" "<<elem.second<<endl;
    }
}