Testcase 3,4 not passing

code: painter problem


I have edited your code, simply I have changed your int values to long long int, rest logic was correct.