NYOJ-诡异的电梯【Ⅰ】-(动态规划)

最新推荐文章于 2020-10-13 16:43:00 发布

陈大美女

最新推荐文章于 2020-10-13 16:43:00 发布

阅读量361

点赞数

分类专栏：动态规划

本文链接：https://blog.csdn.net/chenyang890/article/details/74887855

版权

动态规划专栏收录该内容

18 篇文章 1 订阅

订阅专栏

诡异的电梯【Ⅰ】

时间限制： 1000 ms | 内存限制： 65535 KB

难度： 3

描述

新的宿舍楼有 N(1≤N≤100000) 层 and M(1≤M≤100000)个学生. 在新的宿舍楼里, 为了节约学生的时间也为了鼓励学生锻炼身体, 所以规定该宿舍楼里的电梯在相邻的两层之间是不会连续停下（即，如果在第2层停下就不能在第3层停下。）.所以，如果有学生在相邻的两层之间要停下, 则其中的一部分学生必须选择走楼梯来代替。规定：一个人走下一层楼梯的花费为A，走上一层楼梯的花费为B。（1≤A,B≤100）现在请你设计一个算法来计算出所有学生走楼梯花费的最小费用总和。所有的学生一开始都在第一层，电梯不能往下走，在第二层的时候电梯可以停止。

输入

输入有几组数据T。T(1≤T≤10)
每组数据有N (1≤N≤100000),M(1≤M≤100000),A,B(1≤A,B≤100)。
接下来有M个数字表示每个学生想要停的楼层。

输出

输出看样例。

样例输入

1
3 2 1 1
2 3

样例输出

Case 1: 1

解题思路

这道题是一个一维dp的问题，着眼点在意到第i层楼有两个办法，一种是从i-1层下电梯走楼梯去，或者坐电梯上去，推出状态转移方程dp[i]=min(dp[i-1]+B*a[i],dp[i-2]+mi*a[i-1]);

AC代码

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
int a[100005];
int dp[100005];
int main()
{
    int T;
    scanf("%d",&T);
    int count1=0;
    while(T--)
    {
        int N,M,A,B;
        int mi;
        scanf("%d%d%d%d",&N,&M,&A,&B);
        memset(a,0,sizeof(a));
        memset(dp,0,sizeof(dp));
        mi=min(A,B);
        int x;
        int ans;
        for(int i=1;i<=M;i++)
        {
             scanf("%d",&x);
             a[x]++;
        }
        if(N==1||N==2)
            ans=0;
        else
        {
            for(int i=3;i<=N;i++)
            {
                dp[i]=min(dp[i-1]+B*a[i],dp[i-2]+mi*a[i-1]);

            }
            ans=dp[N];
        }
        printf("Case %d: %d\n",++count1,ans);
    }
    return 0;
}