Zadanie Roznorodność

Question

Siemka wszystkim.

Mam problem z zadaniem Roznorodność z OI: https://szkopul.edu.pl/problemset/problem/eHGwrk9xShVF-z_2f7K4Yyb_/site/?key=statement

Jedyne co udało mi się wymyślić to siłowe rozwarzajycy wszystkie możliwe kwadraty na 10 pkt. Ma ktos pomysł jak lepiej rozwiązać?

EDIT: pozbyłem się mapa i zamieniłem na tablicę (bo limity do 10^5 dla C od i). I przeszło na 40 pkt. Czyli testy n,k,m <= 600.