NC 丢棋子问题

系列文章目录


文章目录

  • 系列文章目录
  • 前言


前言

前些天发现了一个巨牛的人工智能学习网站,通俗易懂,风趣幽默,忍不住分享一下给大家。点击跳转到网站,这篇文章男女通用,看懂了就去分享给你的码吧。
在这里插入图片描述


描述

一座大楼有 n+1 层,地面算作第0层,最高的一层为第 n 层。已知棋子从第0层掉落肯定不会摔碎,从第 i 层掉落可能会摔碎,也可能不会摔碎。

给定整数 n 作为楼层数,再给定整数 k 作为棋子数,返回如果想找到棋子不会摔碎的最高层数,即使在最差的情况下扔的最小次数。一次只能扔一个棋子。
NC 丢棋子问题_第1张图片

class Solution {
public:
    int solve(int N, int K) {
        vector<vector<int>> dp(K+1,vector<int>(N+1,N));
        for(int i=0;i<=K;i++) dp[i][0]=0;        // 层高=0的时候不需要次数
        for(int i=0;i<=N;i++) dp[0][i]=0;        // 棋子=0的时候不需要次数
        for(int i=1;i<=K;i++) dp[i][1]=1;        // 层高=1的时候只需要试1次
        for(int i=1;i<=N;i++) dp[1][i]=i;        // 棋子=1的时候则需要试i次

        for(int i=2;i<=K;i++){
            for(int j=2;j<=N;j++){
                // for(int m=1;m<=j;m++){
                //     dp[i][j]=min(dp[i][j],max(1+dp[i][j-m],1+dp[i-1][m-1]));
                // }

                // 看上面第三重循环的遍历范围是  dp[i-1][0] ~ dp[i-1][j-1] 因此对应上一行选取的dp数组用的是x=0 ~ j-1
                //               所以对应关系为  dp[i][j-1] ~ dp[i][0]
                // x = 0 to j-1 找所有对儿 (每对组合中最大值)的最小值:上一行索引=x,下一行索引=j-1-x
                int l=0;
                int r=j-1;
                while(l<r){                //二分查找
                    int mid=(l+r)/2;
                    int idx=j-1-mid;
                    if(abs(l-r)==1) {dp[i][j]=1+max(dp[i][l],dp[i-1][r]);break;}    // 查找结束
                    if(dp[i-1][mid]==dp[i][idx]) {dp[i][j]=dp[i][idx]+1;break;}     // 直接找到最终结果
                    if(dp[i-1][mid]>dp[i][idx]) {r=mid;continue;}  //如果dp(k-1,mid)大,由于它单增,则权衡值在mid左侧,则r = mid再循环查找 
                    if(dp[i-1][mid]<dp[i][idx]) {l=mid;continue;}  //如果dp(k-1,mid)小,由于它单增,则权衡值在mid右侧,则l = mid再循环查找
                }

            }
        }
        return dp[K][N];
    }
};

你可能感兴趣的:(数据结构)