結果
問題 | No.1437 01 Sort |
ユーザー | tyawanmusi |
提出日時 | 2021-02-20 16:33:14 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,000 ms / 2,000 ms |
コード長 | 2,211 bytes |
コンパイル時間 | 380 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 91,264 KB |
最終ジャッジ日時 | 2024-09-19 12:22:58 |
合計ジャッジ時間 | 13,072 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 37 ms
52,352 KB |
testcase_01 | AC | 38 ms
52,480 KB |
testcase_02 | AC | 38 ms
52,864 KB |
testcase_03 | AC | 34 ms
52,352 KB |
testcase_04 | AC | 38 ms
52,352 KB |
testcase_05 | AC | 35 ms
52,608 KB |
testcase_06 | AC | 34 ms
52,224 KB |
testcase_07 | AC | 35 ms
52,480 KB |
testcase_08 | AC | 35 ms
52,352 KB |
testcase_09 | AC | 39 ms
52,352 KB |
testcase_10 | AC | 34 ms
52,480 KB |
testcase_11 | AC | 36 ms
52,096 KB |
testcase_12 | AC | 37 ms
52,608 KB |
testcase_13 | AC | 906 ms
90,752 KB |
testcase_14 | AC | 909 ms
90,752 KB |
testcase_15 | AC | 346 ms
76,672 KB |
testcase_16 | AC | 827 ms
88,704 KB |
testcase_17 | AC | 524 ms
83,456 KB |
testcase_18 | AC | 821 ms
89,088 KB |
testcase_19 | AC | 661 ms
86,912 KB |
testcase_20 | AC | 533 ms
83,712 KB |
testcase_21 | AC | 942 ms
91,240 KB |
testcase_22 | AC | 985 ms
91,008 KB |
testcase_23 | AC | 942 ms
91,008 KB |
testcase_24 | AC | 1,000 ms
91,264 KB |
testcase_25 | AC | 935 ms
91,136 KB |
testcase_26 | AC | 956 ms
90,624 KB |
ソースコード
# writer 解 # O(N log^2 N) # にぶたんしてくれるやつ from bisect import bisect_left, bisect_right n = int(input()) s = input() ans = 10**10 zero = s.count("0") one = n-zero if zero == 0 or zero == n: exit(print(0)) # 注意事項 out = [0]*n if s[0] == "1": out[0] = 1 for i in range(n-1, -1, -1): if s[i] == "1": out[i] = 1 else: break # "1" である index をとりあえずいっぱい列挙 bit = [] for i in range(n): if s[i] == "1": bit += [i, i+n, i+n+n, i+n+n+n] bit.sort() def f(l, ll): # bit[ll, ll+one) を [l, l+one) に揃えるとき、どれだけグルグル回るか #注意事項 sp_l = (l+one)%n sp_r = n if sp_l == 0: sp_l = n ansl = 0 if bit[ll] <= l+n: # 注意事項 # ((x-1)%n+1) で 0...n-1 ではなく 1...n にする if sp_l <= ((bit[ll]-1)%n+1) <= sp_r and out[bit[ll]%n] == 0: ansl = l+n-bit[ll]-1 else: ansl = l+n-bit[ll] ansr = 0 if l+n+one-1 < bit[ll+one-1]: # "右側" である要素を数えるにぶたん ng = -1 ok = one while ng+1 != ok: mid = (ng+ok)//2 if l+n+mid < bit[ll+mid]: ok = mid else: ng = mid # 注意事項 flag1 = True flag2 = False x = l+n+ok if x%n != 0: x += n-(x%n) if x <= l+n+one-1: if l+n+ok <= bit[ll+ok] <= x: flag2 = True x += n if l+n+ok <= bit[ll+ok] <= x: flag1 = False ansr = one-ok ansr += flag1 ansr -= flag2 return max(ansl, ansr) # 目標の区間を全探索 for l in range(n): # グルグル回る以前の操作回数 anss = (-l-one)%n # 注意事項 sp_l = (l+one)%n sp_r = n if sp_l == 0: sp_l = n # 何回グルグル回れば揃えられるかにぶたん ng = -1 ok = n while ng+1 != ok: mid = (ng+ok)//2 # 回る回数を決め打ちし、bit[ll, ll+one) の ll をできるだけ小さくする # 注意事項 x = l+n-mid-1 if not(sp_l <= ((x-1)%n+1) <= sp_r and out[x%n] == 0): x += 1 ll = bisect_left(bit,x) if f(l, ll) <= mid: ok = mid else: ng = mid ans = min(ans, anss+ok*n) print(ans)