How to optimize code for HackerRank's Fraudulent Activity Notification problem

Question

I have been working to solve this problem on the HackerRank site: Fraudulent Activity Notifications.

Below is the code I have written which satisfies the three sample tes