题意:给出n本书的重量,以及计划要看的书的顺序,一开始书按某种未知顺序垒好,要看哪本书,需要将这本书上面的其他书拿起来,代价为这些书的总重量,看完再将这本书放在最上面,求看完所有书的最小总代价。
题解:贪心
先考虑最先摆放顺序。
对于之前还未出现的书,前面出现的必然在当前书上面,那么最小代价就是在之前的下面。故而顺序就是每本书第一次出现的顺序,然后正常模拟即可。
#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
const int maxn = 1e3 + 5;
int n, m, w[maxn], b[maxn];
set<int> s;
int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) scanf("%d", &w[i]);
for (int i = 1; i <= m; i++) scanf("%d", &b[i]);
int ans = 0;
for (int i = 2; i <= m; i++) {
s.clear();
for (int j = i - 1; j >= 1; j--) {
if (b[i] == b[j]) break;
s.insert(b[j]);
}
set<int>::iterator it = s.begin();
while (it != s.end()) ans += w[*it++];
}
printf("%d\n", ans);
return 0;
}