Codeforces Round #595 (Div. 3) E. By Elevator or Stairs?

题意

n层楼，a[i] (0<i<n)表示从 i 楼到 i + 1 楼走楼梯的时间，b[i] (0<i<n)表示从 i 楼到 i + 1 楼乘电梯的时间，其中每一次乘电梯需要等待 k 时间，楼梯和电梯一次均可上从 x 楼上升到 y 楼 ( y != x )，即一次可以通过楼梯或电梯上升任意层数 。求从1楼到 1 ~ n 层楼所需要的最短时间

思路

dp题：二维数组dp[i][j]，表示通过 j 的方法（ j = 0 表示楼梯，j = 1表示电梯）第 i 层所需的最少时间。
所以得出状态转移方程dp[i][0]=min(dp[i-1][0] + a[i] , dp[i-1][1] + a[i]);
dp[i][1]=min(dp[i-1][1] + b[i], dp[i-1][0] + b[i] + k);
dp[1][0]=0, dp[1][1]=1e18;
答案是min(dp[i][0], dp[i][1])

AC代码

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1e9;
const int N=2e5+10;
ll n,k,a[N],b[N],dp[N][2];
 
int main()
{
	ios::sync_with_stdio(false);
    cin>>n>>k;
    for(int i=1;i<n;i++)
        cin>>a[i];
    for(int i=1;i<n;i++)
        cin>>b[i];
    cout<<0<<" ";
    dp[0][1]=1e18;
    for(int i=1;i<n;i++)
    {
        dp[i][0]=min(dp[i-1][0]+a[i],dp[i-1][1]+a[i]);
        dp[i][1]=min(dp[i-1][1]+b[i],dp[i-1][0]+b[i]+k);
        cout<<min(dp[i][0],dp[i][1])<<" ";
    }
    return 0;
}