modify the func_b(int n, int k)