CF500C New Year Book Reading

Posted 2020-11-13 ppxppx

洛咕

题意:小明非常喜欢读书，他有一共有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
",ans);
    return 0;
}