Skip to content

Commit 980d576

Browse files
author
lucifer
committed
feat: #1227
1 parent 5f32fb1 commit 980d576

File tree

2 files changed

+250
-1
lines changed

2 files changed

+250
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,7 @@ leetcode 题解,记录自己的 leetcode 解题之路。
233233
- [1131.maximum-of-absolute-value-expression](./problems/1131.maximum-of-absolute-value-expression.md) 🆕
234234
- [1186.maximum-subarray-sum-with-one-deletion](./problems/1186.maximum-subarray-sum-with-one-deletion.md) 🆕
235235
- [1218.longest-arithmetic-subsequence-of-given-difference](./problems/1218.longest-arithmetic-subsequence-of-given-difference.md) 🆕
236-
236+
- [1227.airplane-seat-assignment-probability](./problems/1227.airplane-seat-assignment-probability.md) 🆕
237237
- [1261.find-elements-in-a-contaminated-binary-tree](./problems/1261.find-elements-in-a-contaminated-binary-tree.md) 🆕
238238
- [1262.greatest-sum-divisible-by-three](./problems/1262.greatest-sum-divisible-by-three.md) 🆕
239239
- [1297.maximum-number-of-occurrences-of-a-substring](./problems/1297.maximum-number-of-occurrences-of-a-substring.md) 🆕
Lines changed: 249 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,249 @@
1+
## 题目地址(1227. 飞机座位分配概率)
2+
3+
https://leetcode-cn.com/problems/airplane-seat-assignment-probability/description/
4+
5+
## 题目描述
6+
7+
```
8+
9+
有 n 位乘客即将登机,飞机正好有 n 个座位。第一位乘客的票丢了,他随便选了一个座位坐下。
10+
11+
剩下的乘客将会:
12+
13+
如果他们自己的座位还空着,就坐到自己的座位上,
14+
15+
当他们自己的座位被占用时,随机选择其他座位
16+
第 n 位乘客坐在自己的座位上的概率是多少?
17+
18+
 
19+
20+
示例 1:
21+
22+
输入:n = 1
23+
输出:1.00000
24+
解释:第一个人只会坐在自己的位置上。
25+
示例 2:
26+
27+
输入: n = 2
28+
输出: 0.50000
29+
解释:在第一个人选好座位坐下后,第二个人坐在自己的座位上的概率是 0.5。
30+
 
31+
32+
提示:
33+
34+
1 <= n <= 10^5
35+
36+
37+
```
38+
39+
## 暴力递归
40+
41+
这是一道 LeetCode 为数不多的概率题,我们来看下。
42+
43+
### 思路
44+
45+
我们定义原问题为 f(n)。对于第一个人来说,他有 n 中选择,就是分别选择 n 个座位中的一个。由于选择每个位置的概率是相同的,那么选择每个位置的概率应该都是 1 / n。
46+
47+
我们分三种情况来讨论:
48+
49+
- 如果第一个人选择了第一个人的位置(也就是选择了自己的位置),那么剩下的人按照票上的座位做就好了,这种情况第 n 个人一定能做到自己的位置
50+
- 如果第一个人选择了第 n 个人的位置,那么第 n 个人肯定坐不到自己的位置。
51+
- 如果第一个人选择了第 i (1 < i < n)个人的位置,那么第 i 个人就相当于变成了“票丢的人”,此时问题转化为 f(n - i + 1)。
52+
53+
此时的问题转化关系如图:
54+
55+
![](https://tva1.sinaimg.cn/large/006tNbRwly1gb12n0omuuj31bc0ju405.jpg)
56+
(红色表示票丢的人)
57+
58+
整个过程分析:
59+
60+
![](https://tva1.sinaimg.cn/large/006tNbRwly1gb12nhestaj318u0bg76f.jpg)
61+
62+
### 代码
63+
64+
代码支持 Python3:
65+
66+
Python3 Code:
67+
68+
```python
69+
class Solution:
70+
def nthPersonGetsNthSeat(self, n: int) -> float:
71+
if n == 1:
72+
return 1
73+
if n == 2:
74+
return 0.5
75+
res = 1 / n
76+
for i in range(2, n):
77+
res += self.nthPersonGetsNthSeat(n - i + 1) * 1 / n
78+
return res
79+
```
80+
81+
上述代码会栈溢出。
82+
83+
## 暴力递归 + hashtable
84+
85+
### 思路
86+
87+
我们考虑使用记忆化递归来减少重复计算,虽然这种做法可以减少运行时间,但是对减少递归深度没有帮助。还是会栈溢出。
88+
89+
### 代码
90+
91+
代码支持 Python3:
92+
93+
Python3 Code:
94+
95+
```python
96+
class Solution:
97+
seen = {}
98+
99+
def nthPersonGetsNthSeat(self, n: int) -> float:
100+
if n == 1:
101+
return 1
102+
if n == 2:
103+
return 0.5
104+
if n in self.seen:
105+
return self.seen[n]
106+
res = 1 / n
107+
for i in range(2, n):
108+
res += self.nthPersonGetsNthSeat(n - i + 1) * 1 / n
109+
self.seen[n] = res
110+
return res
111+
```
112+
113+
## 动态规划
114+
115+
### 思路
116+
117+
上面做法会栈溢出。其实我们根本不需要运行就应该能判断出栈溢出,题目已经给了数据规模是 1 <= n <= 10 \*\* 5。 这个量级不管什么语言,除非使用尾递归,不然一般都会栈溢出,具体栈深度大家可以查阅相关资料。
118+
119+
既然是栈溢出,那么我们考虑使用迭代来完成。 很容易想到使用动态规划来完成。其实递归都写出来,写一个朴素版的动态规划也难不到哪去,毕竟动态规划就是记录子问题,并建立子问题之间映射而已,这和递归并无本质区别。
120+
121+
### 代码
122+
123+
代码支持 Python3:
124+
125+
Python3 Code:
126+
127+
```python
128+
class Solution:
129+
def nthPersonGetsNthSeat(self, n: int) -> float:
130+
if n == 1:
131+
return 1
132+
if n == 2:
133+
return 0.5
134+
135+
dp = [1, .5] * n
136+
137+
for i in range(2, n):
138+
dp[i] = 1 / n
139+
for j in range(2, i):
140+
dp[i] += dp[i - j + 1] * 1 / n
141+
return dp[-1]
142+
```
143+
144+
这种思路的代码超时了,并且仅仅执行了 35/100 testcase 就超时了。
145+
146+
## 数学分析
147+
148+
### 思路
149+
150+
我们还需要进一步优化时间复杂度,我们需要思考是否可以在线形的时间内完成。
151+
152+
我们继续前面的思路进行分析, 不难得出,我们不妨称其为等式 1:
153+
154+
```
155+
f(n)
156+
= 1/n + 0 + 1/n * (f(n-1) + f(n-2) + ... + f(2))
157+
= 1/n * (f(n-1) + f(n-2) + ... + f(2) + 1)
158+
= 1/n * (f(n-1) + f(n-2) + ... + f(2) + f(1))
159+
```
160+
161+
似乎更复杂了?没关系,我们继续往下看,我们看下 f(n - 1),我们不妨称其为等式 2。
162+
163+
```
164+
f(n-1) = 1/(n-1) * (f(n-2) + f(n-3) + ... + f(1))
165+
```
166+
167+
我们将等式 1 和等式 2 两边分别同时乘以 n 和 n - 1
168+
169+
```
170+
n * f(n) = f(n-1) + f(n-2) + f(n-3) + ... + f(1)
171+
(n-1) * f(n-1) = f(n-2) + f(n-3) + ... + f(1)
172+
```
173+
174+
我们将两者相减:
175+
176+
```
177+
n * f(n) - (n-1)*f(n-1) = f(n-1)
178+
```
179+
180+
我们继续将 (n-1)\*f(n-1) 移到等式右边,得到:
181+
182+
```
183+
n * f(n) = n * f(n-1)
184+
```
185+
186+
也就是说:
187+
188+
```
189+
f(n) = f(n - 1)
190+
```
191+
192+
当然前提是 n 大于 2。
193+
194+
既然如此,我们就可以减少一层循环, 我们用这个思路来优化一下上面的 dp 解法。这种解法终于可以 AC 了。
195+
196+
### 代码
197+
198+
代码支持 Python3:
199+
200+
Python3 Code:
201+
202+
```python
203+
class Solution:
204+
def nthPersonGetsNthSeat(self, n: int) -> float:
205+
if n == 1:
206+
return 1
207+
if n == 2:
208+
return 0.5
209+
210+
dp = [1, .5] * n
211+
212+
for i in range(2, n):
213+
dp[i] = 1/n+(n-2)/n * dp[n-1]
214+
return dp[-1]
215+
```
216+
217+
## 优化数学分析
218+
219+
### 思路
220+
221+
上面我们通过数学分析,得出了当 n 大于 2 时:
222+
223+
```
224+
f(n) = f(n - 1)
225+
```
226+
227+
那么是不是意味着我们随便求出一个 n 就好了? 比如我们求出 n = 2 的时候的值,是不是就知道 n 为任意数的值了。 我们不难想出 n = 2 时候,概率是 0.5,因此只要 n 大于 1 就是 0.5 概率,否则就是 1 概率。
228+
229+
### 代码
230+
231+
代码支持 Python3:
232+
233+
Python3 Code:
234+
235+
```python
236+
class Solution:
237+
def nthPersonGetsNthSeat(self, n: int) -> float:
238+
return 1 if n == 1 else .5
239+
240+
```
241+
242+
## 关键点
243+
244+
- 概率分析
245+
- 数学推导
246+
- 动态规划
247+
- 递归 + mapper
248+
- 栈限制大小
249+
- 尾递归

0 commit comments

Comments
 (0)