non linear top k solution