题意:小明非常喜欢读书,他有一共有n本书,编号为1~n,第i本书重wi。小明计划在暑假的m天里每天读一本书,第i天读第di本书,可能会重复读到同一本书。因为所有的书都是堆成一摞的,所以每次读某本书之前小明都需要先将这本书上面所有的书搬开,拿出这本书,再将搬开的书按原顺序放回去,消耗体力为搬开书的重量之和,读完这本书后将其放在这摞书的最上面。小明想知道这n本书以怎样的初始顺序放置,所搬书消耗总体力最小?\((n<=500,m<=1000)\)
分析:考虑第i次拿书,那么前面i-1次拿的书一定都会产生贡献.换句话说,这里的贡献我们是改变不了,它是个定值.所以我们只能最小化初始序列对操作所产生的贡献.所以初始序列就是每本书第一次被拿出来的顺序了.得到初始序列就直接模拟来做就好了.
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<queue>
#include<map>
#include<set>
#define ll long long
using namespace std;
inline int read(){
int x=0,o=1;char ch=getchar();
while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
if(ch=='-')o=-1,ch=getchar();
while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
return x*o;
}
const int N=1005;
int w[N],a[N],b[N<<1],bj[N];
int main(){
int n=read(),m=read(),cnt=m;
for(int i=1;i<=n;++i)w[i]=read();
for(int i=1;i<=m;++i)a[i]=read();
for(int i=1;i<=m;++i){
if(bj[a[i]])continue;
b[++cnt]=a[i];
bj[a[i]]=1;
}
int l=m+1,ans=0;
for(int i=1;i<=m;++i){
int j=l;
while(a[i]!=b[j]){
ans+=w[b[j]];
++j;
}
b[--l]=a[i];b[j]=0;
}
printf("%d\n",ans);
return 0;
}