Hablu Is in Quarantine

[tophbot]

The soldiers are being trained to fight against the Coronavirus. Mr. Hablu will shake hands with som…

Click here to read the complete problem statement.

---

If you need help solving this problem, mention your approach and ask specific questions. Please avoid sharing your code and asking the Community to figure out “what’s wrong”.

[ForHer03]

I am trying to solve this in python. I get “Accepted” in sample 1 and 2 but “Wrong Answer” on the other.
I need a little hint on what is wrong. Thank you in advance. Here is my code:

Steps:

Take all data in the dictionary along with their id to identify them later
Then take only id of the soldier he will handshake, then if a soldier has ‘c’ in the list, it will print No
Otherwise yes.
To find the people who got infected, find the index of the first person.
#Then identify and put the people who will get infected
id_counter = 1 # Counter for the dictionary

@ForHer03

soldier_data = {} # Store all the numbers in the dictionary
for i in range(int(input())):
soldier_string = input() # Taking them as string to separate them later
for soldiers in soldier_string:
soldier_data[id_counter] = soldiers # Inserting in the dictionary
id_counter += 1 # Increasing the counter to add another soldier

_ = input()

check_list_id = list(map(int, input().split())) # List of the id from user input
check_list =

for check in check_list_id:
check_list.append(soldier_data[check])

will_infect_list =

try: # To avoid an error when c in not found
first_infected = check_list.index(‘c’) # index of the first person who got infected
will_infect_list = check_list[first_infected + 1::] # to find the id of the infected, from first infected to the
# last
except ValueError:
pass

if ‘c’ in check_list: # If there is a infected then it will print No otherwise yes.
print(‘YES’)
else:
print(‘NO’)

for i in sorted(check_list_id[::-1][:len(will_infect_list)][::-1]): # reverse of main list, but took only the length
# amount and then reversed again
print(i, end=’ ')

[md_jakariya]

I have two questions:

1. Will all positive integers (e.g. ID) be distinct (different)?
2. If Mr. Hablu gets affected and then he shakes hands with another corona affected soldier, then should I also print this soldier’s ID?