Counting and equidistribution in Heisenberg groups