top k快排复杂度讨论

面试的时候，被问到top k问题，以为回答维护一个k大小的最小堆就没事了。被面试官引导去思考其他方法，发现居然有快排来搞一搞的邪门方法，但是并没有证明出复杂度为$kO(n)$($k$为常数)。查了像网上的，都是些奇怪的$O(n)+O(n/2)+O(n/4)...<O(2n)$。觉得不大靠谱，决定自己尝试的证明一下。

设$dp(n,k)$表示$n$个数找出前$k$大的时间复杂度。

显然$dp(n,k)=\{\begin{array}{ll}\sum _{i=1}^k\frac{1}{n}(dp(n-i,k-i)+n)+\sum _{i=k+1}^{n-1}\frac{1}{n}(dp(i,k)+n)=n+\frac{1}{n}(\sum _{i=1}^{k-1}dp(n-i,k-i)+\sum _{i=k+1}^{n-1}dp(i,k))& n>k\\ & \\ 0& n\le k\end{array}$

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的结论是错的。接下来来尝试证明一下。

已知

$dp(n,k)=\{\begin{array}{ll}n+\frac{1}{n}(\sum _{i=1}^{k-1}dp(n-i,k-i)+\sum _{i=k+1}^{n-1}dp(i,k))& n>k\\ & \\ 0& n\le k\end{array}$

（得出如果前缀和优化的话求$dp(n,k)$是$O(nk)$的复杂度）

。。。尝试算一下通式，发现是一件非常不靠谱的事情。打印了下，发现$dp(1500,750)$的有1500的3倍多了，感觉应该还是有带$log$的。等有时间or大佬再来算一下吧。