Sri Lanka's slump with the bat on day three cost them dearly, with Australia needing less than 25 minutes to take their final two wickets on Sunday.