よじろめ覚書

私の理解度重視のソースコードです。

ABC140B - Buffet

問題:B - Buffet

#include <iostream>
#include <vector>
using namespace std;

#define REP(i, n) for (int i = 0; i < (n); ++i)

int main(void) {
    int n, ans = 0;

    cin.tie(0);
    ios::sync_with_stdio(false);
    cin >> n;
    vector<int> a(n), b(n), c(n);
    REP(i, n) {
        cin >> a[i];
        a[i]--;
    }
    REP(i, n) {
        cin >> b[i];
    }
    REP(i, n - 1) {
        cin >> c[i];
    }
    REP(i, n) {
        ans += b[a[i]];
        if (0 < i && a[i] == a[i - 1] + 1) {
            ans += c[a[i - 1]];
        }
    }

    cout << ans << "\n";
    return 0;
}