51nod 1306 高楼和棋子 dp

题意

有个N层的高楼和若干个棋子，所有的棋子都是一样的。棋子从楼的某层E扔到地上不会碎(0 <= E <= N)，但从比这个楼层高的地方扔到地上都会碎。给出楼的高度N，以及棋子的数量M，你来找出这个E(0 <= E <= N)，问最坏情况下需要实验多少次才能计算出准确的E（如果棋子摔碎了，就不能继续用这个棋子进行测试了）。
1 <= T <= 50000，1 <= N <= 10^18, 1 <= M <= 64。

分析

这题是04年的论文题，文中提到了很多种优化方法。

朴素的想法是设f[i,j]表示i层楼用j个棋子最坏情况下需要多少步，转移是f[i,j]=min(max(f[w-1,j-1],f[i-w,j])+1)。复杂度是$O(n^3)$。
注意到当没有蛋数量的限制时，答案就是二分查找的最坏情况，是$log_2n$。也就是说如果$j>log_2n$就直接输出就好了。复杂度变成了$O(n^2log_2n)$。
根据定义不难发现有$f[i-1,j]<=f[i,j]$，那么转移的时候决策点的贡献是一个单峰函数，可以三分，由于坐标均为整数，所以可以二分。复杂度变成了$O(n(log_2n)^2)$。
再推一下不难发现这个转移有决策单调性。复杂度变成了$O(nlog_2n)$。

这已经是优化到不能再优化了，然而还是不能通过本题，于是只能考虑另外的做法。

设f[i,j]表示有i个棋子扔了j次最坏情况下可以计算出多少楼。
转移时f[i,j]=f[i-1,j-1]+f[i,j-1]+1
为什么是这样呢？考虑第一次扔到某一个位置w，如果棋子坏了那么就取剩下的，就是f[i-1,j-1]；否则就可以继续往上算f[i,j-1]层。显然w取f[i-1,j-1]+1时是最优的，所以还要+1。
边界条件：f[n,1]=1,f[1,n]=n。根据归纳法不难得出f[2,n]=n*(n+1)/2。
注意到$C_j^i<=f[i,j]$，所以有用的状态其实并不多，可以全部预处理出来然后二分就好了。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
#include<cmath>
using namespace std;

typedef long long LL;

const int N=2000005;
const LL MX=1e18;

LL n;
int m;
vector<LL> f[65];

int main()
{
    for (int i=0;i<=N;i++) f[0].push_back(0);
    for (int i=1;i<=64;i++)
    {
        f[i].push_back(0);
        for (int j=1;j<=N;j++)
        {
            f[i].push_back(f[i][j-1]+f[i-1][j-1]+1);
            if (f[i][j]>MX) break;
        }
    }
    int T;scanf("%d",&T);
    while (T--)
    {
        scanf("%lld%d",&n,&m);
        if (m==1) printf("%lld\n",n);
        else if (m==2)
        {
            LL x=sqrt(n*2);
            if (x*(x+1)<n*2) x++;
            printf("%lld\n",x);
        }
        else
        {
            int x=lower_bound(f[m].begin(),f[m].end(),n)-f[m].begin();
            printf("%d\n",x);
        }
    }
    return 0;
}