結果
問題 |
No.1701 half price
|
ユーザー |
![]() |
提出日時 | 2021-02-22 09:49:23 |
言語 | PyPy3 (7.3.15) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,718 bytes |
コンパイル時間 | 358 ms |
コンパイル使用メモリ | 82,220 KB |
実行使用メモリ | 76,244 KB |
最終ジャッジ日時 | 2024-07-23 03:26:39 |
合計ジャッジ時間 | 6,950 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 16 WA * 4 |
ソースコード
class INPUT: def __init__(self): from sys import argv if argv[-1] in ["./Main.py","prog.py","Main.py"]:#Atcoder:./Main.py,Wandbox:prog.py,yukicoder:Main.py self.l=open(0).read().split()[::-1] else: self.l=open("_input.txt",mode="r").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 def change(n,k): ret="" while n!=0: ret+=str(n%k) n//=k return ret def main(): N,W=pin(2) assert(1<=N<=13) assert(0<=W<=pow(10,9)) A=pin(0) for i in A: assert(0<=i<=pow(10,9)) debug=i%2 assert(debug==0) ans=set() B=[i//2 for i in A] for i in range(3**N): s=0 b=0 x=1 u=change(i,3) for j in range(len(u)): if u[j]=="1": s+=A[j] b+=x elif u[j]=="2": s+=B[j] b+=x x*=2 if s==W: ans.add(b) print(len(ans)) return main()