面试的时候,被问到top k问题,以为回答维护一个k大小的最小堆就没事了。被面试官引导去思考其他方法,发现居然有快排来搞一搞的邪门方法,但是并没有证明出复杂度为 k O ( n ) kO(n) kO(n)( k k k为常数)。查了像网上的,都是些奇怪的 O ( n ) + O ( n / 2 ) + O ( n / 4 ) . . . < O ( 2 n ) O(n)+O(n/2)+O(n/4)...<O(2n) O(n)+O(n/2)+O(n/4)...<O(2n)。觉得不大靠谱,决定自己尝试的证明一下。
设 d p ( n , k ) dp(n,k) dp(n,k)表示 n n n个数找出前 k k k大的时间复杂度。
显然 d p ( n , k ) = { ∑ i = 1 k 1 n ( d p ( n − i , k − i ) + n ) + ∑ i = k + 1 n − 1 1 n ( d p ( i , k ) + n ) = n + 1 n ( ∑ i = 1 k − 1 d p ( n − i , k − i ) + ∑ i = k + 1 n − 1 d p ( i , k ) ) n > k 0 n ≤ k dp(n,k)=\begin{cases}\displaystyle\sum_{i=1}^k\frac{1}{n}(dp(n-i,k-i)+n)+\displaystyle\sum_{i=k+1}^{n-1}\frac{1}{n}(dp(i,k)+n)=n+\frac{1}{n}(\displaystyle\sum_{i=1}^{k-1}dp(n-i,k-i)+\displaystyle\sum_{i=k+1}^{n-1}dp(i,k))&{n>k}\\\\0&{n\le k}\end{cases} dp(n,k)=⎩⎪⎪⎪⎪⎨⎪⎪⎪⎪⎧i=1∑kn1(dp(n−i,k−i)+n)+i=k+1∑n−1n1(dp(i,k)+n)=n+n1(i=1∑k−1dp(n−i,k−i)+i=k+1∑n−1dp(i,k))0n>kn≤k
ok,dp式子推出来了,打个简单的暴力康康。
#include<bits/stdc++.h>
using namespace std;
const int maxn = 20;
double dp[maxn][maxn];
int main() {
for(int k = 1; k <= maxn-5; k++)
for(int n = 1; n <= maxn-5; n++) {
if (n <= k) {
dp[n][k] = 0;
continue;
}
dp[n][k] = 0;
for (int i = 1; i < k; i++) {
dp[n][k] += dp[n-i][k-i];
}
for (int i = k+1; i < n; i++){
dp[n][k] += dp[i][k];
}
dp[n][k] /= n;
dp[n][k] += n;
}
for (int n = 1;n <= maxn-5; n++) {
printf("n:%d=>", n);
for (int k = 1; k <= maxn-5; k++) {
printf("%.4f ",dp[n][k]);
if (k == maxn-5) {
cout << endl;
}
}
}
return 0;
}
结果如图
显然,存在许多不止2n的地方。明显关于时间复杂度小于2n的结论是错的。接下来来尝试证明一下。
已知
d p ( n , k ) = { n + 1 n ( ∑ i = 1 k − 1 d p ( n − i , k − i ) + ∑ i = k + 1 n − 1 d p ( i , k ) ) n > k 0 n ≤ k dp(n,k)=\begin{cases}n+\frac{1}{n}(\displaystyle\sum_{i=1}^{k-1}dp(n-i,k-i)+\displaystyle\sum_{i=k+1}^{n-1}dp(i,k))&{n>k}\\\\0&{n\le k}\end{cases} dp(n,k)=⎩⎪⎪⎪⎪⎨⎪⎪⎪⎪⎧n+n1(i=1∑k−1dp(n−i,k−i)+i=k+1∑n−1dp(i,k))0n>kn≤k
(得出如果前缀和优化的话求
d
p
(
n
,
k
)
dp(n,k)
dp(n,k)是
O
(
n
k
)
O(nk)
O(nk)的复杂度)
。。。尝试算一下通式,发现是一件非常不靠谱的事情。打印了下,发现 d p ( 1500 , 750 ) dp(1500,750) dp(1500,750)的有1500的3倍多了,感觉应该还是有带 l o g log log的。等有时间or大佬再来算一下吧。