https://ac.nowcoder.com/acm/contest/11168/B
思路:
反向考虑,那么从后往前拿的顺序就是个栈的顺序,1先放,2在1的上面,3在2的上面。这个顺序同样也是最优的。这个顺序也就是按输入的顺序。
考虑这么贪心为什么是对的呢?尝试把两个相邻位置换了,可以发现移动的总重量和不会变小。
模拟的一个细节:
都知道碰到原来的出现过的数就break了。但是要注意把原来出现过的数只算一次,不然下面的数可能就会多算同一个编号的球。(天知道这里没想到wa了多久)
#include<iostream>
#include<vector>
#include<queue>
#include<cstring>
#include<cmath>
#include<map>
#include<set>
#include<cstdio>
#include<algorithm>
#define debug(a) cout<<#a<<"="<<a<<endl;
using namespace std;
const int maxn=2e3+100;
typedef long long LL;
inline LL read(){LL x=0,f=1;char ch=getchar(); while (!isdigit(ch)){if (ch=='-') f=-1;ch=getchar();}while (isdigit(ch)){x=x*10+ch-48;ch=getchar();}
return x*f;}
LL a[maxn],tot=0;
LL val[maxn];
int main(void)
{
cin.tie(0);std::ios::sync_with_stdio(false);
LL n,m;cin>>n>>m;
for(LL i=1;i<=n;i++) cin>>val[i];
LL sum=0;
for(LL i=1;i<=m;i++){
LL x;cin>>x;
bool flag=1;
multiset<LL>s;
for(LL j=tot;j>=1;j--){
if(s.count(a[j])) continue;
s.insert(a[j]);
if(a[j]==x){
flag=0;
a[++tot]=x;
break;
}
else sum+=val[a[j]];
}
if(flag==1) a[++tot]=x;
}
cout<<sum<<"\n";
return 0;
}