How to apply a function on each row of a Spark Dataframe after groupby using Java

Question

I\'m using Spark 3.0.x in Java and we\'ve a transformation requirement that goes as follows:

A sample (hypothetical) Dataframe:

+=====