题目简述
某互联网公司一年一度的春招开始了,一共有 n 名面试者入选。每名面试者都会提交一份简历,公司会根据提供的简历资料产生一个预估的能力值,数值越大代表越有可能通过面试。
小 A 和小 B 负责审核面试者,他们均有所有面试者的简历,并且将各自根据面试者能力值从大到小的顺序浏览。由于简历事先被打乱过,能力值相同的简历的出现顺序是从它们的全排列中等可能地取一个。现在给定 n 名面试者的能力值 scores,设 X 代表小 A 和小 B 的浏览顺序中出现在同一位置的简历数,求 X 的期望。
提示:离散的非负随机变量的期望计算公式为 E(X) = Σ(k=1, ∞) kPr(X = k)。在本题中,由于 X 的取值为 0 到 n 之间,期望计算公式可以是E(X) = Σ(k=1, n) kPr(X = k)。
题解示例
示例 1:
输入:scores = [1,2,3]
输出:3
解释:由于面试者能力值互不相同,小 A 和小 B 的浏览顺序一定是相同的。X的期望是 3 。
示例 2:
输入:scores = [1,1]
输出:1
解释:设两位面试者的编号为 0, 1。由于他们的能力值都是 1,小 A 和小 B 的浏览顺序都为从全排列 [[0,1],[1,0]] 中等可能地取一个。如果小 A 和小 B 的浏览顺序都是 [0,1] 或者 [1,0] ,那么出现在同一位置的简历数为 2 ,否则是 0 。所以 X 的期望是 (2+0+2+0) * 1/4 = 1
示例 3:
输入:scores = [1,1,2]
输出:2
数据范围
1 <= scores.length <= 10^5
0 <= scores[i] <= 10^6
标记难度
难度:简单
通过次数:6051
提交次数:8963
通过率:67.5%
问题解析
数组中的元素可以分为两种:单元素和多元素。
对于单元素,由于两位面试官都是根据能力值从大到小的顺序浏览,假设数组元素按数值非递c增的顺序排列,那么其在数组中的顺序一定是相同的,即单元素在两位面试官的浏览顺序中一定是出现在同一位置的,则单元素的数学期望为1。
对于多元素,假设该元素出现m次,编号为0,1,2,…,m-1。假设m = 3,则全排列有如下6种情况:(0,1,2),(0,2,1),(1,0,2),(1,2,0),(2,0,1),(2,1,0),现两个面试官分别从六种顺序里挑选一种,易知每个位置编号相同的概率均为1/3,那么可以得出两位面试官在同一位置选取同一元素的概率为1/n,则多元素的数学期望为1/n × n = 1。
可以得出,无论是单元素还是多元素,其数学期望均为1,即整个数组的数学期望等于数组中数值不同的元素的个数和。
python3代码
class Solution:
def expectNumber(self, scores: List[int]) -> int:
return len(set(scores))
C语言代码
int expectNumber(int* scores, int scoresSize){
int hash[1000000] = {0};
int result = 0;
for (int i = 0; i < scoresSize; ++i) {
if (hash[scores[i]] == 0) {
hash[scores[i]]++;
result++;
}
}
return result;
}
C++代码
class Solution {
public:
int expectNumber(vector<int>& scores) {
return set<int>(scores.begin(), scores.end()).size();
}
};
大佬专属代码
# 暂时没看到任何“出类拔萃”的代码
敲黑板!!!
这道题关键在于看懂题目,只能说:概率论yyds!!!