結果
問題 |
No.1470 Mex Sum
|
ユーザー |
![]() |
提出日時 | 2021-04-09 22:55:49 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 248 ms / 2,000 ms |
コード長 | 1,629 bytes |
コンパイル時間 | 96 ms |
コンパイル使用メモリ | 12,800 KB |
実行使用メモリ | 28,280 KB |
最終ジャッジ日時 | 2024-06-25 06:39:08 |
合計ジャッジ時間 | 11,441 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 49 |
ソースコード
class INPUT: def __init__(self): self.l=open(0).read().split()[::-1] self.length=len(self.l) return def stream(self,k=1,f=int,f2=False): assert(-1<k) m=self.length if m==0 or m<k: raise Exception("There is no input!") elif f!=str: if k==0: self.length=0 return list(map(f,self.l[::-1])) if k==1 and not f2: self.length-=1 return f(self.l.pop()) if k==1 and f2: self.length-=1 return [f(self.l.pop())] ret=[] for _ in [0]*k: ret.append(f(self.l.pop())) self.length-=k return ret else: if k==0: self.length=0 return self.l[::-1] if k==1 and not f2: self.length-=1 return self.l.pop() if k==1 and f2: self.length-=1 return [self.l.pop()] ret=[] for _ in [0]*k: ret.append(self.l.pop()) self.length-=k return ret pin=INPUT().stream """ pin(number[default:1],f[default:int],f2[default:False]) if number==0 -> return left all listを変数で受け取るとき、必ずlistをTrueにすること。 """ def main(): N=pin(1) A=pin(0) d1=[0] d2=[0] for i in A: if i==1: d1.append(d1[-1]+1) else: d1.append(d1[-1]) if i==2: d2.append(d2[-1]+1) else: d2.append(d2[-1]) ans=0 for i in range(N): s1=d1[-1]-d1[i]#1の個数 s2=d2[-1]-d2[i]#2の個数 t=N-i-1-s1-s2#それ以外の個数 if A[i]==1: ans+=s1*2+s2*3+t*2 elif A[i]==2: ans+=s1*3+s2+t else: ans+=s1*2+s2+t print(ans) return main()