[javascript]力扣题库之库存管理

库存管理

仓库管理员以数组 stock 形式记录商品库存表，其中 stock[i] 表示对应商品库存余量。请返回库存余量最少的 cnt 个商品余量，返回 顺序不限。

示例 1：

输入：stock = [2,5,7,4], cnt = 1
输出：[2]
示例 2：

输入：stock = [0,2,3,6], cnt = 2
输出：[0,2] 或 [2,0]

提示：

0 <= cnt <= stock.length <= 10000

0 <= stock[i] <= 10000

解法一：

var inventoryManagement = function(stock, cnt) {
	return stock.sort((a, b) => a-b).slice(0, cnt)
}

解法二: 计数排序，将空间转时间

var inventoryManagement = function(stock, cnt) {
	let bucket = []
    let sortedIndex = 0

    for (let i = 0; i < stock.length; i++) { // 把stock中元素值当做下标计数
        if (!bucket[stock[i]]) {             // 例如stock=[2,5,7,4]
            bucket[stock[i]] = 0			 // 则bucket为[0,0,1,0,1,1,0,1]
        }
        bucket[stock[i]]++
    }

    let result = []
    for (let j = 0; j < bucket.length; j++) {
        while (bucket[j]-- > 0 && sortedIndex < cnt) {
            result[sortedIndex] = j
            sortedIndex++
        }
        if (sortedIndex === cnt) {
            break
        }
    }
    return result
}