結果

問題 No.2549 Paint Eggs
コンテスト
ユーザー Nzt3
提出日時 2024-01-11 09:29:34
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 75 ms / 2,000 ms
コード長 2,220 bytes
コンパイル時間 2,500 ms
コンパイル使用メモリ 250,140 KB
実行使用メモリ 9,036 KB
最終ジャッジ日時 2024-09-27 20:25:46
合計ジャッジ時間 6,721 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 45
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
using namespace std;
using ll=long long;
namespace Lib {
template <class S, auto op, auto e>
struct segtree {
  int size, _n;
  vector<S> seg;
  segtree(int n) {
    _n = n;
    size = 1;
    while (size < n) size <<= 1;
    seg = vector<S>(size * 2, e());
  }
  void set(int p, S val) {
    p += size;
    seg[p] = val;
    while (p /= 2) {
      seg[p] = op(seg[p * 2], seg[p * 2 + 1]);
    }
  }
  S prod(int l, int r) {
    S retL = e(), retR = e();
    for (l += size, r += size; l < r; l /= 2, r /= 2) {
      if (l % 2) retL = op(retL, seg[l++]);
      if (r % 2) retR = op(seg[--r], retR);
    }
    return op(retL, retR);
  }
  S get(int p) { return seg[p + size]; }
  template <class F>
  int max_right(int L, F f) {
    L += size;
    S sm = e();
    do {
      while (L % 2 == 0) L /= 2;
      if (!f(op(sm, seg[L]))) {
        while (L < size) {
          L *= 2;
          if (f(op(sm, seg[L]))) {
            sm = op(sm, seg[L]);
            L++;
          }
        }
        return L - size;
      }
      sm = op(sm, seg[L]);
      L++;
    } while ((L & -L) != L);
    return _n;
  }

  template <class F>
  int min_left(int R, F f) {
    R += size;
    S sm = e();
    do {
      R--;
      while (R > 1 && R % 2 == 1) R /= 2;
      if (!f(op(seg[R], sm))) {
        while (R < size) {
          R = R * 2 + 1;
          if (f(op(seg[R], sm))) {
            sm = op(seg[R], sm);
            R--;
          }
        }
        return R + 1 - size;
      }
      sm = op(seg[R], sm);
    } while ((R & -R) != R);
    return 0;
  }
};
}  // namespace Lib
int main(){
  ios::sync_with_stdio(false);
  cin.tie(nullptr);
  int N,M;
  ll K;
  cin>>N>>M>>K;
  vector C(N,0),A(M,0);
  for(int &i:C)cin>>i,--i;
  for(int &i:A)cin>>i;
  auto seg_op=[](ll a,ll b){
    return min(a,b);
  };
  auto seg_e=[](){
    return (ll)1e18;
  };
  Lib::segtree<ll,seg_op,seg_e> seg(M);
  for(int i=0;i<M;i++){
    seg.set(i,A[i]*K);
  }
  for(int i=0;i<K;i++){
    seg.set(C[i],seg.get(C[i])-A[C[i]]);
  }
  ll ans=seg.prod(0,M);
  for(int i=K;i<N;i++){
    seg.set(C[i-K],seg.get(C[i-K])+A[C[i-K]]);
    seg.set(C[i],seg.get(C[i])-A[C[i]]);
    ans=min(ans,seg.prod(0,M));
  }
  cout<<ans<<'\n';
}
0