算法-买票问题

算法-买卖股票问题

8月还没发博客,来一个动态规划练练手。

本题目来源是网易笔试题

题目描述:现在有n个人排队买票,已知是早上八点开始卖票,这n个人买票有两种方式
第一种是每个人都可以单独去买自己的票,第i个人花费i秒。
第二种是每个人都可以选择和后面的人一起买票,第i个人和第i+1个人一共花费b[i]秒。
最后一个人只能和前一个一起买票或者自己单独购买,求花费的最短时间。

可以看到,买票问题是一个典型的动态规划问题,为什么这样说?因为当前最短时间依赖于之前的买票决策。

确定了是动态规划问题后,我们的主要任务就是寻找状态转移方程了。

以下面的例子做个说明


        int[] a={
    
    3,5,7,6};
        int[] b={
    
    6,11,13};

在解决这个问题时,我们实际上是在求在数组a中,到从位置0到位置ptr处,我们累计的最小值

例如,当ptr=0时,显然当前最短时间时间

dp[ptr]=a[0]

当ptr=1时,当前最短时间为

dp[ptr]=min(a[0]+a[1],b[0])

显然,当ptr=2时,最短时间为min(dp[ptr-1]+a[ptr],dp[ptr-2]+b[ptr-1])。

dp[ptr]=min(dp[ptr-1]+a[ptr],dp[ptr-2]+b[ptr-1])

后面的都可以继续递推

于是,我们的出这样的结论

    /**
     * @param a 单独购买的时间
     * @param b 两个人一起买的时间
     * @return
     */
    public static int timeToClose(int[] a,int[] b){
    
    
        if (a.length==0||a.length==1){
    
    
            return a.length==0?0:a[0];
        }
        int[] dp=new int[a.length];
        dp[0]=a[0];
        dp[1]=Math.min(b[0],a[0]+a[1]);
        int ptr=2;
        while (ptr<a.length){
    
    
            //a[ptr]表示单独购买的花费时间,dp[ptr-1]表示从第0个人到第ptr-1个人总共花费的最小时间
            //b[ptr-1]表示第ptr个人和第ptr-1个人一起花费的时间,dp[ptr-2]表示从第0个人到第ptr-2个人总共花费的最小时间
            dp[ptr++]=Math.min(a[ptr]+dp[ptr-1],b[ptr-1]+dp[ptr-2]);
        }
        return dp[dp.length-1];
    }

猜你喜欢

转载自blog.csdn.net/qq_23594799/article/details/107882361
今日推荐