SWEA 9700 USB 꽂기의 미스터리(with Python)

내가 생각한 Solution

문제에서 생각해볼 점

• $s_1$: 한번 뒤집고 꽂는 케이스

  • $s_1=(1-p) \cdot q$

• $s_2$: 두번 뒤집고 꽂는 케이스

  • $s_2=p \cdot (1-q) \cdot q$

코드 구현

T = int(input())
 
output = ''
for tc in range(1, T+1):
    p, q = map(float, input().split())
 
    s1 = (1-p)*q
    s2 = p*(1-q)*q
 
    if s1 < s2:
        output += f'#{tc} YES\n'
    else:
        output += f'#{tc} NO\n'
 
print(output)