Spark Scala - add new column to dataframe/data by conditionally checking number of other coliumns

Question

Here is the scenario from legacy code need to converted to spark scala. Any pointers will be highly appreciated.

Scenario: Need to add a new column to a dataframe/dataset