Sì, al posto del break; in quel modo appena hai fatto lo scambio esci dalla funzione - il che ha senso, visto che, fatto uno scambio, non resta più nulla da fare.