[SPARK-2554][SQL] CountDistinct partial aggregation and object allocation improvements