广州4399面试题（一）

广州4399面试题（一）

By 马冬亮(凝霜  Loki)

一个人的战争(http://blog.csdn.net/MDL13412)

题目描述

现给定一个含有n个元素的数组，请随机获取其中的m个元素(不能重复获取)。

算法描述

首先，随机获取元素，可以使用rand() % 数组长度；

其次，要保证元素的不重复获取，只需将获取的元素从原数组中移除即可，但是每次都进行删除操作，需要频繁的移动数组元素，其复杂度很高；现在，我们换一种思路，将获取的元素与原数组最后的元素进行交换，再将数组的长度减一，那么就可以做到O(1)复杂度将其移除；详见下图：

此算法的时间复杂度为O(n)，其只与要获取的m个数有关，在n和m相差非常悬殊的时候，效率非常高。

程序源码

#include <iostream>
#include <vector>
#include <algorithm>
#include <iterator>

using namespace std;

typedef int ErrorType;

const ErrorType ErrSucceed          =   0;
const ErrorType ErrInvalidRange     =   10;
const ErrorType ErrUnknown          =   100;

template <typename ItemType>
ErrorType GetRandItemsFromArray(vector<ItemType> &items,
                                const size_t itemCount, 
                                vector<ItemType> &result)
{
    // 边界检测
    if (items.size() < itemCount)
        return ErrInvalidRange;

    // 将结果向量清空，并预留足够的空间，防止多次分配导致的性能开销。
    result.clear();
    result.reserve(itemCount);

    // 注意差1的边界错误
    size_t itemsLength = items.size() - 1;

    // 算法描述见正文
    for (size_t i = 0; i < itemCount; ++i)
    {
        int tmpIndex = rand() % itemsLength;
        result.push_back(items[tmpIndex]);
        swap(items[tmpIndex], items[itemsLength]);
        --itemsLength;
    }

    return ErrSucceed;
}

int main()
{
    vector<int> vec;
    for (int i = 0; i < 100; ++i)
        vec.push_back(i);
    vector<int> res;
    for (int i = 0; i < 10; ++i)
    {
        GetRandItemsFromArray(vec, 10, res);
        copy(res.begin(), res.end(), ostream_iterator<int>(cout, " "));
        cout << endl;
    }

    return 0;
}

总结

题目本身不难，但是要注意边界的校验，错误处理，代码可读性，数据抽象，时间及空间复杂度。另外，如果是现场写代码，最好问面试官原始数组是否允许修改！