Submission #4778289


Source Code Expand

Copy
B,W=map(int,input().split())
N=B+W+2
mod=10**9+7
table=[1]*(N+3)
t=1
for i in range(1,N+3):
t*=i
t%=mod
table[i]=t
rtable=[1]*(N+3)
t=1
for i in range(1,N+3):
t*=pow(i,mod-2,mod)
t%=mod
rtable[i]=t
p=[0]*(B+W+1)
q=[0]*(B+W+1)
t=pow(2,mod-2,mod)
for i in range(W,B+W+1):
if i==W:
p[i]=pow(t,i,mod)
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
B,W=map(int,input().split())
N=B+W+2
mod=10**9+7
table=[1]*(N+3)
t=1
for i in range(1,N+3):
    t*=i
    t%=mod
    table[i]=t
rtable=[1]*(N+3)
t=1
for i in range(1,N+3):
    t*=pow(i,mod-2,mod)
    t%=mod
    rtable[i]=t
p=[0]*(B+W+1)
q=[0]*(B+W+1)
t=pow(2,mod-2,mod)
for i in range(W,B+W+1):
    if i==W:
        p[i]=pow(t,i,mod)
    else:
        p[i]=(p[i-1]+table[i-1]*rtable[W-1]*rtable[i-W]*pow(t,i,mod))%mod
            #(table[i]*rtable[W]*table[i-W]*pow(t,i,mod))%mod
for i in range(B,B+W+1):
    if i==B:
        q[i]=pow(t,i,mod)
    else:
        q[i]=(q[i-1]+table[i-1]*rtable[B-1]*rtable[i-B]*pow(t,i,mod))%mod
#print(p,q)
for i in range(B+W):
    ans=((1+p[i]-q[i])*pow(2,mod-2,mod))%mod
    print(ans)

Submission Info

Submission Time
Task E - Black or White
User okumura
Language PyPy3 (2.4.0)
Score 700
Code Size 751 Byte
Status AC
Exec Time 845 ms
Memory 79640 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 700 / 700
Status
AC × 3
AC × 15
Set Name Test Cases
Sample sample_01.txt, sample_02.txt, sample_03.txt
All sample_01.txt, sample_02.txt, sample_03.txt, test_01.txt, test_02.txt, test_03.txt, test_04.txt, test_05.txt, test_06.txt, test_07.txt, test_08.txt, test_09.txt, test_10.txt, test_11.txt, test_12.txt
Case Name Status Exec Time Memory
sample_01.txt AC 169 ms 38256 KB
sample_02.txt AC 170 ms 38256 KB
sample_03.txt AC 169 ms 38256 KB
test_01.txt AC 827 ms 78544 KB
test_02.txt AC 584 ms 66072 KB
test_03.txt AC 845 ms 79640 KB
test_04.txt AC 598 ms 66060 KB
test_05.txt AC 775 ms 74012 KB
test_06.txt AC 520 ms 59196 KB
test_07.txt AC 510 ms 59852 KB
test_08.txt AC 664 ms 70092 KB
test_09.txt AC 670 ms 70088 KB
test_10.txt AC 557 ms 64352 KB
test_11.txt AC 556 ms 64348 KB
test_12.txt AC 445 ms 58240 KB


2025-04-08 (Tue)
21:58:19 +00:00