Improve test cases, even O(n) solution passes


#1

My O(n) solution got accepted, test cases are very weak.