結果
問題 | No.884 Eat and Add |
ユーザー | None |
提出日時 | 2021-03-02 02:13:23 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 113 ms / 1,000 ms |
コード長 | 2,082 bytes |
コンパイル時間 | 252 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 110,080 KB |
最終ジャッジ日時 | 2024-10-03 01:26:56 |
合計ジャッジ時間 | 1,747 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 41 ms
52,736 KB |
testcase_01 | AC | 40 ms
52,608 KB |
testcase_02 | AC | 40 ms
52,480 KB |
testcase_03 | AC | 113 ms
105,856 KB |
testcase_04 | AC | 108 ms
104,704 KB |
testcase_05 | AC | 112 ms
103,040 KB |
testcase_06 | AC | 110 ms
103,680 KB |
testcase_07 | AC | 69 ms
87,040 KB |
testcase_08 | AC | 71 ms
87,296 KB |
testcase_09 | AC | 101 ms
110,080 KB |
testcase_10 | AC | 41 ms
52,224 KB |
testcase_11 | AC | 42 ms
52,736 KB |
ソースコード
class Compress: def __init__(self,array): """ A=[(a,degeneracy),...] """ S=[0] self.array=array[:] self.A=[] for a, cnt in array: S.append(S[-1]+cnt) self.A.append(a) self.S=S[1:] def __getitem__(self, i): j=bisect_left(self.S,i+1) return self.A[j] def sum(self, i): """ sum in [0,i) """ j=bisect_left(self.S,i) res=0 q=0 for k in range(j): res+=self.A[k]*(self.S[k]-q) q=self.S[k] print(j, res) if j>0: res+=self.A[j]*(i-self.S[j-1]) else: res+=self.A[j]*(i) return res def sort(self): S=[0] self.A=[] for a, cnt in sorted(self.array): S.append(S[-1]+cnt) self.A.append(a) self.S=S[1:] def __iter__(self): l=0 for i in range(self.S[-1]): if i==self.S[l]: l+=1 yield self.A[l] def __str__(self): return " ".join(list(map(str, self))) def deg(A): """ 連続して続く文字を圧縮する(※Counterではない) [1,1,1,2,2,2,3] -> [(1,3),(2,3),(3,1)] """ res=[] a0, cnt=A[0], 1 for a in A[1:]+[None]: if a!=a0: res.append((a0,cnt)); cnt=1 else: cnt+=1 a0=a return res def bit_rep(bit,N): return format(bit, "0" + str(N) + "b") def bits_rep(bits,N): return [format(bit, "0" + str(N) + "b") for bit in bits] ############################################################### import sys from bisect import * input = sys.stdin.readline N=input().rstrip() S=list(map(int,N)) data=deg(S) res=0 i=0 A=[] while i<len(data): if i>=len(data)-2 or data[i][0]==0 or data[i][1]<=1: A.append(data[i]) else: j=i+1 cnt=data[i][1] while j<len(data) and data[j][1]==1: res+=1 cnt+=1+data[j+1][1] j+=2 i=j-1 A.append((1,cnt)) i+=1 for a,cnt in A: if a==0:continue res+=min(2,cnt) print(res)