1.剑指 Offer II 071. 按权重生成随机数
给定一个正整数数组 w ,其中 w[i] 代表下标 i 的权重(下标从 0 开始),请写一个函数 pickIndex ,它可以随机地获取下标 i,选取下标 i 的概率与 w[i] 成正比。
例如,对于 w = [1, 3],挑选下标 0 的概率为 1 / (1 + 3) = 0.25 (即,25%),而选取下标 1 的概率为 3 / (1 + 3) = 0.75(即,75%)。
也就是说,选取下标 i 的概率为 w[i] / sum(w) 。
题目大意:按照给定的权重来生成随机数
解题思路:前缀和+二分,首先,计算出前缀数组的一个前缀和数组,而前缀和数组的最后一个值,也就是整个权重数组的总权重,而每一项权重和总权重的比值,就是该权重所代表下表的“概率”,然后总权重值作为随机数范围,然后二分随机值的位置,即可按照权重生成相应的随机数
代码:
/**
* @param {number[]} w
*/
var Solution = function(w) {
this.sums = [w[0]]
for(let i = 1; i < w.length; i ++) {
this.sums[i] = this.sums[i - 1] + w[i]
}
};
/**
* @return {number}
*/
Solution.prototype.pickIndex = function() {
let l = 0, r = this.sums.length - 1;
const p = Math.random() * this.sums[r]
while(l <= r) {
const mid = l + ((r - l) >> 1)
if(this.sums[mid] > p) {
if(mid === 0 || this.sums[mid - 1] <= p) {
return mid
}
r = mid - 1;
} else {
l = mid + 1;
}
}
return -1;
};
2.剑指 Offer II 073. 狒狒吃香蕉
狒狒喜欢吃香蕉。这里有 n 堆香蕉,第 i 堆中有 piles[i] 根香蕉。警卫已经离开了,将在 h 小时后回来。
狒狒可以决定她吃香蕉的速度 k (单位:根/小时)。每个小时,她将会选择一堆香蕉,从中吃掉 k 根。如果这堆香蕉少于 k 根,她将吃掉这堆的所有香蕉,然后这一小时内不会再吃更多的香蕉,下一个小时才会开始吃另一堆的香蕉。
狒狒喜欢慢慢吃,但仍然想在警卫回来前吃掉所有的香蕉。
返回她可以在 h 小时内吃掉所有香蕉的最小速度 k(k 为整数)。
题目大意:求一个最小值k,能够确保数组piles被分为h等分k
解题思路:二分k的值,知道求出最小值。
代码:
/**
* @param {number[]} piles
* @param {number} h
* @return {number}
*/
const check = (piles, k) => {
let res = 0;
for(const item of piles) {
res += Math.floor(item / k);
if(item % k) res ++;
}
return res;
}
var minEatingSpeed = function(piles, h) {
let l = 0, r = 0;
for(const item of piles) {
r = Math.max(r, item)
}
let res = 0;
while(l <= r) {
const m = l + ((r - l) >> 1)
const temp = check(piles, m)
if(temp <= h) {
res = m;
r = m - 1;
} else {
l = m + 1;
}
}
return res;
};