結果
問題 | No.1838 Modulo Straight |
ユーザー | pockyny |
提出日時 | 2022-02-12 22:27:05 |
言語 | C++17 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 1,182 ms / 2,000 ms |
コード長 | 2,001 bytes |
コンパイル時間 | 1,011 ms |
コンパイル使用メモリ | 90,056 KB |
実行使用メモリ | 88,024 KB |
最終ジャッジ日時 | 2023-09-11 11:00:15 |
合計ジャッジ時間 | 29,950 ms |
ジャッジサーバーID (参考情報) |
judge11 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,336 KB |
testcase_01 | AC | 2 ms
5,568 KB |
testcase_02 | AC | 2 ms
5,408 KB |
testcase_03 | AC | 4 ms
5,744 KB |
testcase_04 | AC | 4 ms
5,568 KB |
testcase_05 | AC | 4 ms
5,444 KB |
testcase_06 | AC | 4 ms
5,516 KB |
testcase_07 | AC | 5 ms
5,548 KB |
testcase_08 | AC | 5 ms
5,704 KB |
testcase_09 | AC | 5 ms
5,536 KB |
testcase_10 | AC | 6 ms
5,764 KB |
testcase_11 | AC | 1,106 ms
87,848 KB |
testcase_12 | AC | 1,173 ms
87,944 KB |
testcase_13 | AC | 1,182 ms
87,800 KB |
testcase_14 | AC | 1,169 ms
87,920 KB |
testcase_15 | AC | 1,152 ms
88,024 KB |
testcase_16 | AC | 988 ms
49,464 KB |
testcase_17 | AC | 1,018 ms
49,464 KB |
testcase_18 | AC | 1,040 ms
49,396 KB |
testcase_19 | AC | 992 ms
43,488 KB |
testcase_20 | AC | 964 ms
43,652 KB |
testcase_21 | AC | 986 ms
43,484 KB |
testcase_22 | AC | 346 ms
35,140 KB |
testcase_23 | AC | 365 ms
30,676 KB |
testcase_24 | AC | 365 ms
15,780 KB |
testcase_25 | AC | 436 ms
14,916 KB |
testcase_26 | AC | 475 ms
15,044 KB |
testcase_27 | AC | 574 ms
15,104 KB |
testcase_28 | AC | 550 ms
13,524 KB |
testcase_29 | AC | 650 ms
15,052 KB |
testcase_30 | AC | 687 ms
16,012 KB |
testcase_31 | AC | 729 ms
18,176 KB |
testcase_32 | AC | 780 ms
23,128 KB |
testcase_33 | AC | 981 ms
49,284 KB |
testcase_34 | AC | 970 ms
49,324 KB |
testcase_35 | AC | 866 ms
30,340 KB |
testcase_36 | AC | 794 ms
22,144 KB |
testcase_37 | AC | 685 ms
15,280 KB |
testcase_38 | AC | 597 ms
13,788 KB |
testcase_39 | AC | 225 ms
30,084 KB |
testcase_40 | AC | 2 ms
5,432 KB |
ソースコード
#include <iostream> #include <vector> #include <atcoder/lazysegtree> using namespace std; using namespace atcoder; typedef long long ll; ll inf = 1000000000000000000; struct S{ll x,sz;}; struct F{ll x;}; S op(S l,S r){return S{l.x + r.x,l.sz + r.sz};} S e(){return S{0,0};} S mapping(F l,S r){return S{r.x + l.x*r.sz,r.sz};} F compo(F l,F r){return F{l.x + r.x};} F id(){return F{0};} ll a[400010],b[400010]; vector<ll> solve(vector<int> &v,int k){ int i,m = v.size(); vector<int> pos(m); for(i=0;i<m;i++) pos[v[i]] = i; vector<S> ve(m); for(i=0;i<m;i++) ve[i] = {0,1}; lazy_segtree<S,op,e,F,mapping,compo,id> seg(ve),seg2(ve); vector<ll> ret(m); for(i=0;i<m;i++){ ll inv = seg.prod(v[i] + 1,m).x; seg2.apply(i,i + 1,F{inv}); seg.apply(v[i],v[i] + 1,F{1}); } ret[0] = seg2.all_prod().x; for(i=1;i<m;i++){ ll inv = seg2.get(pos[i - 1]).x; seg2.apply(pos[i - 1],pos[i - 1] + 1,F{-inv}); seg2.apply(pos[i - 1] + 1,m,F{1}); ret[i] = seg2.all_prod().x; } return ret; } int main(){ int i,j,m,k; cin >> m >> k; for(i=0;i<m*k;i++) cin >> a[i]; vector<vector<int>> ind(m); for(i=0;i<m*k;i++) ind[a[i]].push_back(i); for(i=0;i<m;i++){ for(j=0;j<k;j++) b[ind[i][j]] = j; } /*for(i=0;i<m*k;i++) cout << b[i] << " "; cout << endl;*/ ll ans = 0; vector<S> ve(k); for(i=0;i<k;i++) ve[i] = {0,1}; lazy_segtree<S,op,e,F,mapping,compo,id> seg(ve); for(i=0;i<m*k;i++){ ans += seg.prod(b[i] + 1,k).x; seg.apply(b[i],b[i] + 1,F{1}); } vector<vector<int>> v(k); for(i=0;i<m*k;i++) v[b[i]].push_back(a[i]); vector<ll> sum(m); for(i=0;i<k;i++){ vector<ll> nv = solve(v[i],k); /*for(j=0;j<m;j++) cout << nv[j] << " "; cout << endl;*/ for(j=0;j<m;j++) sum[j] += nv[j]; } ll mx = inf; for(i=0;i<m;i++) mx = min(mx,sum[i]); ans += mx; cout << ans << endl; return 0; }