Axon Software Engineer 2 interview questions

This was for technical screen. Question was not too difficult (med), hashmap + binary search based. Was able to suggested multiple approaches including best time and space complexity. Coded in time. Wrote test cases. Interviewer mentioned one edge case that I missed and I changed logic to handle it but it caused some other edge cases to be missed. Interviewer then suggested a change to add which also did not work. We went over time and couldn't find why certain things were not returning the correct output. This was not great from the interviewer and I would have thought their lack of direction wouldn't count towards the rest of the approach that I presented. Though, the application came back rejected. Going to leave some feedback for them, but this just feels.. unexpected.