My friends solve it with dicnic.
But its time complexity can be up to O(nm^sqrt(n+m)),and its space compexity can be up to O(nm).
I think admin should add the test that all the bank belong to one country.