結果
問題 | No.1437 01 Sort |
ユーザー |
![]() |
提出日時 | 2021-02-20 00:26:48 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,104 bytes |
コンパイル時間 | 131 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 33,652 KB |
最終ジャッジ日時 | 2024-09-19 01:13:01 |
合計ジャッジ時間 | 4,052 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 10 TLE * 1 -- * 13 |
ソースコード
n=int(input())s=input()ans=10**10zero=s.count("0")one=n-zeroif zero==0 or zero==n:exit(print(0))se=set()if s[0]=="1":se.add(0)for i in range(n-1,-1,-1):if s[i]=="1":se.add(i)else:breakpo=[]for i in range(n):if s[i]=="1":po+=[i,i+n,i+n+n,i+n+n+n]po.sort()for l in range(n):a=list(range(l+n,l+one+n))sp_l=(l+one)%nsp_r=nif sp_l==0:sp_l=nfor ll in range(one*3):ansl=0ansr=0flag=Trueflagg=Falseif po[ll]<=a[0]:if sp_l<=((po[ll]-1)%n+1)<=sp_r and (po[ll]%n not in se):ansl=a[0]-po[ll]-1else:ansl=a[0]-po[ll]if a[one-1]<po[ll+one-1]:flag=Trueflagg=Falseng=-1ok=onewhile ng+1!=ok:mid=(ng+ok)//2if a[mid]<po[ll+mid]:ok=midelse:ng=midx=a[ok]if x%n:x+=n-(x%n)if x<=a[one-1]:if a[ok]<=po[ll+ok]<=x:flagg=Truex+=nif a[ok]<=po[ll+ok]<=x:flag=Falseansr=one-okif ansr:ansr+=flagansr-=flaggans=min(ans,max(ansl,ansr)*n+(-l-one)%n)print(ans)