Lead/lag in R but only for rows that meet a yes/no condition

Question

I have a dataset of patient visits that looks like this:

   visit infection treatment
1      1  negative         1
2      2  negative         1
3      3  posi