結果
問題 | No.2549 Paint Eggs |
ユーザー |
![]() |
提出日時 | 2023-11-25 14:04:11 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,895 bytes |
コンパイル時間 | 3,369 ms |
コンパイル使用メモリ | 260,796 KB |
実行使用メモリ | 47,104 KB |
最終ジャッジ日時 | 2024-09-26 10:38:03 |
合計ジャッジ時間 | 24,208 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
other | AC * 43 WA * 2 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/lazysegtree> using namespace std; using ll=long long; using pll=pair<ll,ll>; using tll=tuple<ll,ll,ll>; using ld=long double; const ll INF=(1ll<<60); #define rep(i,n) for(ll i=0;i<(ll)(n);i++) #define all(v) v.begin(),v.end() template<class T> void chmin(T &a,T b){ if(a>b){ a=b; } } template<class T> void chmax(T &a,T b){ if(a<b){ a=b; } } struct s{ __int128 v,k,a; }; bool operator<(s a,s b){ return a.v<b.v; } s op(s a,s b){ return min(a,b); } s e(){ return {INF,INF,INF}; } s mapping(ll f,s x){ s ret=x; ret.k+=f; ret.v=ret.a*ret.k; return ret; } ll composition(ll f,ll g){ return f+g; } ll id(){ return 0; } int main(){ ios::sync_with_stdio(false); cin.tie(nullptr); ll n,m,k; cin >> n >> m >> k; vector<vector<ll>> v(m); vector<ll> c(n); rep(i,n){ cin >> c[i]; c[i]--; } vector<ll> a(m); rep(i,m) cin >> a[i]; atcoder::lazy_segtree<s,op,e,ll,mapping,composition,id> seg(m); rep(i,m) seg.set(i,{0,0,a[i]}); rep(i,k){ seg.apply(0,c[i],1); seg.apply(c[i]+1,m,1); } ll ans=seg.all_prod().v; for(ll i=k;i<n;i++){ seg.apply(0,c[i-k],-1); seg.apply(c[i-k]+1,m,-1); seg.apply(0,c[i],1); seg.apply(c[i]+1,m,1); chmin(ans,(ll)seg.all_prod().v); } vector<ll> cnt(m,0); rep(i,n) cnt[c[i]]++; vector<ll> ord(m); rep(i,m) ord[i]=i; sort(all(ord),[&](ll i,ll j){ return cnt[j]<cnt[i]; }); ll q=0; for(auto &i:ord){ ll now=0; rep(j,k){ if(c[j]!=i) now+=a[i]; } chmin(ans,now); for(ll j=k;j<n;j++){ if(c[j-k]!=i) now-=a[i]; if(c[j]!=i) now+=a[i]; } q++; if(1000<=q) break; } cout << ans << '\n'; }