PAT(甲级) 1008 Elevator

题目

题目链接

思路

比较当前层和将要去的层，如果下一层大于所在层，就作差乘6，否则下一层小于当前层，作差乘4，当然最后还得加上每层停留的5秒

代码

#include <iostream>
#include <vector>
using namespace std;
int main(){
    int cnt = 0, data, ans = 0;
    cin >> cnt;
    vector<int> vec;
    for(int i=0; i<cnt; ++i){
        cin >> data;
        vec.push_back(data);
    }
    ans += vec[0]*6;
    for(int i=1; i<vec.size(); ++i){
        if(vec[i] > vec[i-1])
            ans = ans + (vec[i]-vec[i-1])*6;
        else
            ans = ans + (vec[i-1]-vec[i])*4;
    }
    cout << ans+cnt*5;
    return 0;
}