问题

春节期间小明使用微信收到很多个红包,非常开心。在查看领取红包记录时发现,某个红包金额出现的次数超过了红包总数的一半。请帮小明找到该红包金额。写出具体算法思路和代码实现,要求算法尽可能高效。

给定一个红包的金额数组gifts及它的大小n,请返回所求红包的金额。

若没有金额超过总数的一半,返回0。

测试样例:

[1,2,3,2,2],5
返回:2

解决方案

方法一:C++中学过map的人可能做这个题会非常简单,利用map键值对的巧妙来对数据的处理。以下就是根据map<int,int>的第一个int用来存储被计数的值,右边来存放出现的个数。

class Gift {
public:
    int getValue(vector<int> gifts, int n) {
        map<int ,int> m;
        for(int i=0;i<n;++i)
        {
            m[gifts[i]]++;
        }
        for(auto e:m)
        {
            if(e.second > n/2) //second就是m的第二个参数,即第二个int
            {
                return e.first;  //同理,first是第一个int了

            }           
        }
        return 0;
    }
};

方法二:当然,非常了解C++库中函数的人可能会想到一个非常简单的方法,利用STL中的算法count函数计数功能,三两行代码就能解决这个问题。利用出现次数超过总元素个数一半知道这个数一定会是中位数,当然,前提是这个数组有序。我们看下面的具体代码。


class Gift {
public:
    int getValue(vector<int> gifts, int n) {
        sort(gifts.begin(),gifts.end());
        int ret = count(gifts.begin(),gifts.end(),gifts[n/2]);
        if(ret > n/2)
        {
            return gifts[n/2];
        }
        return 0;
    }
};

方法三:在上面两种特殊方式做出的基础上,我们思考了用普通的算法来求解该问题,我们的思路是首先对数组里的数进行一个判断,找出出现次数最多的那个数,然后再对它进行计数判断。以下就是具体代码。

class Gift {
public:
    int getValue(vector<int> gifts, int n) {
        int gift = gifts[0];
        int count = 1;
         
        for(int i = 1; i < n; ++i)
        {
            if(gifts[i] == gift)
            {
                count++;
            }
            else
            {
                count--;
            }
            if(count == 0)
            {
                gift = gifts[i];
                count = 1;
            }
        }
        count = 0;
        for(int i = 0; i < n ;++i)
        {
            if(gifts[i] == gift)
            {
                count++;
            }
        }
        if(count > n/2){
            return gift;
        }
        return 0;
    }
};