Codeforces Good Bye 2014 C問題 - New Year Book Reading

Source

Codeforces Good Bye 2014 C問題 (1000pt)
Problem description

問題概要

$N$ 冊の本があって,本 $k$ の重さは $W_k$ である.
この中で,これから順番に読みたい本が $M$ 冊与えられる.
読みたい本を読む際には,その本の上に積まれている本を持ち上げ,読みたい本を取り,読み終わったら一番上に置く.
この際,持ち上げた本の重さの和だけ労力がかかる.
本は縦に積み上がっていて,最初はどのように積んでいても良いとする.(労力の和が最小になるようになっているとする)
必要な労力の和の最小値を求める問題.

解法

まだ $1$ 回も読んでいない本については,初めて読むときに一番都合が良い場所にあったと解釈して良い.
(もしくは,最初早く読む順番に上から順番に並んでいたと思っても良い)
すると,各本を読むとき,その本が前回読まれたのより後に読まれたことのある本の分だけ持ち上げれば良い.
後は,各本について一番最後に読んだ時刻を覚えつつ,愚直にシミュレーションすれば良い.
時間計算量は $O(NM)$ 程度.

C++のスパゲッティなコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define READER_BUF_SIZE 1048576
#define WRITER_BUF_SIZE 1048576
int reader_pt=READER_BUF_SIZE,reader_last;char reader_buf[READER_BUF_SIZE];
int writer_pt=0;char writer_buf[WRITER_BUF_SIZE];
#define mygc(c) {if(reader_pt==READER_BUF_SIZE)reader_pt=0,reader_last=fread(reader_buf,sizeof(char),READER_BUF_SIZE,stdin);(c)=reader_buf[reader_pt++];}
#define mypc(c) {if(writer_pt==WRITER_BUF_SIZE)writer_pt=0,fwrite(writer_buf,sizeof(char),WRITER_BUF_SIZE,stdout);writer_buf[writer_pt++]=(c);}
#define myed {fwrite(writer_buf,sizeof(char),writer_pt,stdout);writer_pt=0;}

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
template <class T, class S> void reader(T *x, S *y){reader(x);reader(y);}

void writer(int x, char c){int s=0,m=0;char f[10];if(x<0)m=1,x=-x;while(x)f[s++]=x%10,x/=10;if(!s)f[s++]=0;if(m)mypc('-');while(s--)mypc(f[s]+'0');mypc(c);}
template<class T> void writerLn(T x){writer(x,'\n');}

int N, D;
int W[2000], B[2000];
int used[2000];

int main(){
  int i, j, k;
  int res = 0;

  reader(&N,&D);
  rep(i,N) reader(W+i);
  rep(i,D) reader(B+i), B[i]--;

  rep(i,N) used[i] = -1;
  rep(i,D){
    rep(j,N) if(used[j] > used[B[i]]) res += W[j];
    used[B[i]] = i;
  }

  writerLn(res);

  myed;
  return 0;
}

Current time: 2017年07月21日13時26分50秒
Last modified: 2015年01月11日08時31分20秒 (by laycrs)
Tags: Competitive_Programming Codeforces Codeforces_Good_Bye_2014
トップページに戻る

Logged in as: unknown user (not login)

ログイン: