COVID RUN PROBLEM, Unable to complete first sub-task

help
1

try:
tc = int(input())
cd = [[] for x in range(0, tc)]
infected = []
for i in range(0, tc):
cd[i] = input().split()

for i in range(0, tc):
    for j in range(0, len(cd[i])):
        cd[i][j] = int(cd[i][j])

for i in range(0, tc):
    infected.append(cd[i][2])
    new = [x for x in range(0, cd[i][0])]
    temp = cd[i][2]
    for j in range(0, int(cd[i][1])):
        temp = (temp + cd[i][1]) % cd[i][0]
        infected.append(temp)
    if cd[i][3] in infected:
        print("YES")
    else:
        print("NO")

except:
pass

2

hopefully this code may help you for better understanding
#include
using namespace std;

int main() {
int n; cin>>n;
while(n–)
{
int n,k,x,y,j=0; cin>>n>>k>>x>>y;
if(y>n)
cout<<“NO”<<endl;
else
for(int i=1;i<=n;i++)
{
if((x+k*i)%n==y)
{
cout<<“YES”<<endl;
j=1;
break;
}
}
if(j==0)
cout<<“NO”<<endl;

}
return 0;
}