Đếm năm cần để thăng cấp quân hàm bằng C++

Mô tả bài toán

Quân đội n cấp bậc (1..n). Mảng D[n-1]: D[i] năm từ cấp i lên i+1. Amal ở cấp a, muốn lên cấp b. Tính tổng năm cần.

Ví dụ

D=[5,6], a=1, b=3 → 5+6=11 năm.

Thuật toán

Tổng D[i] cho i từ a-1 đến b-2.

Code C++

#include <bits/stdc++.h>
using namespace std;

int solve(vector D, int a, int b) {
    int sum = 0;
    for (int i = a-1; i < b-1; i++)
        sum += D[i];
    return sum;
}

int main() {
    vector D = {5,6};
    int a=1,b=3;
    cout << solve(D,a,b) << endl;
    return 0;
}

Đầu vào / Đầu ra

Input: {5,6}, 1, 3

Output: 11