贝壳找房 2020校招笔试 算法工程师笔试题 采木 锯子斧子砍树

题目要求,在你面前依次排着n棵树,而你有两种工具,锯子和斧头,他们砍第i棵树的时间分别为ai和bi,一开始你拿的是斧头,而砍第i棵树前交换工具需要花费ci的时间,问题来了,依次砍完这些树的时间最短为多少呢?

输入

第一行一个数n(n<=100)
接下来n行,每行三个数ai,bi,ci(i<=30000)

输出

一个数最短时间

样例输入

3
20 40 20
10 4 25
90 100 5

样例输出

139

分析:这个题是状态转移的dp,跟这个比较相近的有leetcode股票的题目中,有一种用状态转移同时解掉这三道股票题的题解。

具体分析在代码注释中体现

#include
#include 
#include 
using namespace std;

int main()
{
    int n;
    cin>>n;
    vector a(n+1), b(n+1), c(n+1);
    for(int i=1;i<=n;i++)
    {
        cin>>a[i]>>b[i]>>c[i];
    }
    vector> dp;   //第二维斧子是1, 锯子是0
    dp.resize(n+1, vector(2));
    dp[1][1] = b[1];         //砍第1棵树用斧子的时间
    dp[1][0] = a[1] + c[1];  //砍第1棵树用锯子的时间,还需要换一下工具
    for(int i=2; i<=n;i++)
    {
        dp[i][1] = min(dp[i-1][1], dp[i-1][0] + c[i]) + b[i];
        dp[i][0] = min(dp[i-1][0], dp[i-1][1] + c[i]) + a[i];
    }
    cout<

 

你可能感兴趣的:(编程入门)