Help me in solving PRIMEREVERSE problem

help
,
1

My issue

runtime error is there
output is correct what i do

My code

# cook your dish here
# cook your dish here
def P(lst1,lst2):
    if(lst1.count(1)==lst2.count(1)):
        print("YES")
    else:
        print("NO")
for _ in range(int(input())):
    x=int(input())
    lst1=list(map(int,str(int(input()))))
    lst2=list(map(int,str(int(input()))))
    P(lst1,lst2)

Learning course: Roadmap to 3*
Problem Link: https://www.codechef.com/learn/course/klu-roadmap-3star/KLURMP300A/problems/PRIMEREVERSE