某笔试：一道贪心

题目

对一个序列可以操作任意次，每一次操作可以让某一个位置上的数加1，但是每一个位置的数最多只能操作一次。通过这个操作能让新产生的序列中最多有多少个不同的数。

输入: nums = [1, 2, 2, 2, 5, 6]
输出: 5
说明: 序列变为[1, 2, 3, 2, 5, 6]，此时不同元素最多

思路

先将数组排序。

使用贪心算法，贪心就贪在仅对与前一个元素相等的元素进行加1操作

遍历数组，使用变量pre记录前一个数组元素，按照当前数组元素与pre的关系分为三种情况：

• 当前数组元素等于pre：当前数组元素加1，不同元素个数加1，pre记录当前数组元素
• 当前数组元素小于pre：跳过，什么也不干
• 当前数组元素大于pre：不同元素个数加1，pre记录当前数组元素

代码

int getDifferent(vector& nums)
{
    sort(nums.begin(), nums.end());
    int sz = nums.size();
    
    int pre = nums[0];
    int count = 1;
    for (int i = 1; i < sz; ++i)
    {
        if (nums[i] == pre)
        {
            ++nums[i];
            ++count;
            pre = nums[i]
        }
        else if (nums[i] < pre)
        {
            continue;
        }
        else
        {
            ++count;
            pre = nums[i];
        }
    }
    
    return count;
}