Make sure you are comfortable in writing graph based code easily , questions are modified and you need to identify the pattern, write complex code and run the test cases. I was able to write the code and run all test cases.