Extract IDs based on difference in timestamp- spark

Question

I have a spark dataframe that has two columns ("time_stamp" and "ID").

Example dataframe:

      **ID**                **time_stamp*